方法一
直接使用最原始的方式导入:
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