torch.nn.Transformer导入失败

方法一

直接使用最原始的方式导入：

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