当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python torch.nn.TransformerDecoder用法及代码示例


用法:

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

参数

  • decoder_layer-TransformerDecoderLayer() 类的实例(必需)。

  • num_layers-解码器中sub-decoder-layers 的数量(必需)。

  • norm-层标准化组件(可选)。

TransformerDecoder 是 N 个解码器层的堆栈

例子::
>>> decoder_layer = nn.TransformerDecoderLayer(d_model=512, nhead=8)
>>> transformer_decoder = nn.TransformerDecoder(decoder_layer, num_layers=6)
>>> memory = torch.rand(10, 32, 512)
>>> tgt = torch.rand(20, 32, 512)
>>> out = transformer_decoder(tgt, memory)

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.TransformerDecoder。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。