直接使用最原始的方式导入:
from torch.nn.modules.transformer import Transformer, TransformerEncoder, TransformerDecoder,
TransformerEncoderLayer, TransformerDecoderLayer
方法二
在torch.nn.modules文件夹下新建一个_init_.pyi文件,然后写入这段语句:
from .transformer import TransformerEncoder as TransformerEncoder, TransformerDecoder as TransformerDecoder,
TransformerEncoderLayer as TransformerEncoderLayer, TransformerDecoderLayer as TransformerDecoderLayer,
Transformer as Transformer
此时就可以正常使用:
from torch.nn import Transformer, TransformerEncoder, TransformerDecoder,
TransformerEncoderLayer, TransformerDecoderLayer



