[参数完全一致]torch.nn.TransformerDecoder

torch.nn.TransformerDecoder

torch.nn.TransformerDecoder(decoder_layer, num_layers, norm=None)

paddle.nn.TransformerDecoder

paddle.nn.TransformerDecoder(decoder_layer, num_layers, norm=None)

其中功能一致, 参数完全一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
decoder_layer decoder_layer TransformerDecoderLayer 的一个实例。
num_layers num_layers TransformerDecoderLayer 层的叠加数量。
norm norm 层标准化(Layer Normalization)。