[参数完全一致]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)。 |