torch. nn. Transformer import failed

Method 1

Import directly in the most primitive way ：

from torch.nn.modules.transformer import Transformer, TransformerEncoder, TransformerDecoder, \
    TransformerEncoderLayer, TransformerDecoderLayer

Method 2

stay torch.nn.modules Create a new folder _init_.pyi file , Then write this statement ：

from .transformer import TransformerEncoder as TransformerEncoder, TransformerDecoder as TransformerDecoder, \
    TransformerEncoderLayer as TransformerEncoderLayer, TransformerDecoderLayer as TransformerDecoderLayer,  \
    Transformer as Transformer

At this time, it can be used normally ：

from torch.nn import Transformer, TransformerEncoder, TransformerDecoder, \
    TransformerEncoderLayer, TransformerDecoderLayer