nn.ModuleDict |
部分支持 |
暂不支持图模式 |
nn.Parameter |
支持 |
|
nn.UninitializedParameter |
部分支持 |
暂不支持图模式 |
nn.UninitializedBuffer |
部分支持 |
暂不支持图模式 |
nn.ParameterList |
部分支持 |
暂不支持图模式 |
nn.ParameterDict |
部分支持 |
暂不支持图模式 |
nn.Unfold |
支持 |
|
nn.Fold |
支持 |
|
nn.MaxPool1d |
支持 |
|
nn.MaxPool2d |
支持 |
|
nn.MaxPool3d |
支持 |
|
nn.AvgPool1d |
支持 |
|
nn.AvgPool2d |
支持 |
|
nn.AvgPool3d |
支持 |
|
nn.FractionalMaxPool2d |
支持 |
|
nn.FractionalMaxPool3d |
支持 |
|
nn.LPPool1d |
部分支持 |
功能存在限制 |
nn.LPPool2d |
部分支持 |
功能存在限制 |
nn.AdaptiveMaxPool1d |
部分支持 |
功能存在限制 |
nn.AdaptiveMaxPool2d |
部分支持 |
功能存在限制 |
nn.AdaptiveMaxPool3d |
支持 |
|
nn.AdaptiveAvgPool1d |
支持 |
|
nn.AdaptiveAvgPool2d |
支持 |
|
nn.AdaptiveAvgPool3d |
支持 |
|
nn.ReflectionPad1d |
支持 |
|
nn.ReflectionPad2d |
支持 |
|
nn.ReflectionPad3d |
部分支持 |
功能存在限制 |
nn.ReplicationPad1d |
支持 |
|
nn.ReplicationPad2d |
支持 |
|
nn.ReplicationPad3d |
支持 |
|
nn.ZeroPad2d |
支持 |
|
nn.ConstantPad1d |
支持 |
|
nn.ConstantPad2d |
支持 |
|
nn.ConstantPad3d |
支持 |
|
nn.ELU |
部分支持 |
功能存在限制 |
nn.Hardshrink |
部分支持 |
功能存在限制 |
nn.Hardsigmoid |
支持 |
|
nn.Hardtanh |
部分支持 |
功能存在限制 |
nn.Hardswish |
部分支持 |
功能存在限制 |
nn.LeakyReLU |
部分支持 |
功能存在限制 |
nn.LogSigmoid |
支持 |
|
nn.PReLU |
部分支持 |
功能存在限制 |
nn.ReLU |
支持 |
|
nn.ReLU6 |
部分支持 |
功能存在限制 |
nn.RReLU |
部分支持 |
inplace不支持图模式 |
nn.SELU |
部分支持 |
inplace不支持图模式 |
nn.CELU |
部分支持 |
inplace不支持图模式 |
nn.GELU |
支持 |
|
nn.Sigmoid |
支持 |
|
nn.SiLU |
支持 |
|
nn.Mish |
部分支持 |
功能存在限制 |
nn.Softplus |
支持 |
|
nn.Softshrink |
部分支持 |
功能存在限制 |
nn.Softsign |
支持 |
|
nn.Tanh |
支持 |
|
nn.Tanhshrink |
支持 |
|
nn.Threshold |
部分支持 |
inplace不支持图模式 |
nn.GLU |
支持 |
|
nn.Softmin |
支持 |
|
nn.Softmax |
支持 |
|
nn.Softmax2d |
支持 |
|
nn.LogSoftmax |
部分支持 |
功能存在限制 |
nn.BatchNorm1d |
支持 |
|
nn.BatchNorm2d |
支持 |
|
nn.BatchNorm3d |
支持 |
|
nn.GroupNorm |
支持 |
|
nn.LayerNorm |
支持 |
|
nn.LocalResponseNorm |
支持 |
|
nn.RNNBase |
支持 |
|
nn.RNN |
部分支持 |
功能存在限制 |
nn.RNNCell |
支持 |
|
nn.LSTMCell |
支持 |
|
nn.GRUCell |
支持 |
|
nn.Identity |
支持 |
|
nn.Linear |
部分支持 |
功能存在限制 |
nn.Bilinear |
支持 |
|
nn.Dropout |
部分支持 |
inplace 不支持图模式 |
nn.Dropout1d |
部分支持 |
inplace 不支持图模式 |
nn.Dropout2d |
部分支持 |
inplace 不支持图模式 |
nn.Dropout3d |
部分支持 |
inplace 不支持图模式 |
nn.AlphaDropout |
部分支持 |
inplace 不支持图模式 |
nn.FeatureAlphaDropout |
部分支持 |
inplace 不支持图模式 |
nn.CosineSimilarity |
支持 |
|
nn.PairwiseDistance |
支持 |
|
nn.L1Loss |
支持 |
|
nn.MSELoss |
支持 |
|
nn.CrossEntropyLoss |
部分支持 |
功能存在限制 |
nn.CTCLoss |
支持 |
|
nn.NLLLoss |
支持 |
|
nn.PoissonNLLLoss |
支持 |
|
nn.GaussianNLLLoss |
支持 |
|
nn.BCELoss |
支持 |
|
nn.BCEWithLogitsLoss |
支持 |
|
nn.MarginRankingLoss |
支持 |
|
nn.HingeEmbeddingLoss |
支持 |
|
nn.HuberLoss |
支持 |
|
nn.SmoothL1Loss |
支持 |
|
nn.SoftMarginLoss |
部分支持 |
暂不支持CPU后端 |
nn.MultiLabelSoftMarginLoss |
支持 |
|
nn.CosineEmbeddingLoss |
支持 |
|
nn.TripletMarginWithDistanceLoss |
支持 |
|
nn.PixelShuffle |
支持 |
|
nn.PixelUnshuffle |
支持 |
|
nn.Upsample |
部分支持 |
功能存在限制 |
nn.UpsamplingNearest2d |
部分支持 |
功能存在限制 |
nn.UpsamplingBilinear2d |
支持 |
|
nn.ChannelShuffle |
支持 |
|
nn.Flatten |
支持 |
|
nn.Unflatten |
支持 |
|
nn.Module |
部分支持 |
功能存在限制 |
nn.Sequential |
支持 |
|
nn.ModuleList |
部分支持 |
功能存在限制 |
nn.Conv1d |
部分支持 |
功能存在限制 |
nn.Conv2d |
部分支持 |
功能存在限制 |
nn.Conv3d |
部分支持 |
功能存在限制 |
nn.ConvTranspose1d |
部分支持 |
功能存在限制 |
nn.ConvTranspose2d |
部分支持 |
功能存在限制 |
nn.ConvTranspose3d |
支持 |
|
nn.MaxUnpool1d |
支持 |
|
nn.MaxUnpool2d |
支持 |
|
nn.MaxUnpool3d |
支持 |
|
nn.MultiheadAttention |
部分支持 |
功能存在限制 |
nn.AdaptiveLogSoftmaxWithLoss |
部分支持 |
暂不支持图模式 |
nn.SyncBatchNorm |
部分支持 |
仅支持Ascend后端 |
nn.InstanceNorm1d |
部分支持 |
仅支持GPU后端 |
nn.InstanceNorm2d |
部分支持 |
仅支持GPU后端 |
nn.InstanceNorm3d |
部分支持 |
仅支持GPU后端 |
nn.LSTM |
部分支持 |
功能存在限制 |
nn.GRU |
部分支持 |
功能存在限制 |
nn.Embedding |
部分支持 |
功能存在限制 |
nn.KLDivLoss |
支持 |
|
nn.MultiLabelMarginLoss |
部分支持 |
暂不支持CPU后端 |
nn.MultiMarginLoss |
支持 |
|
nn.TripletMarginLoss |
支持 |
|
nn.Transformer |
部分支持 |
功能存在限制 |
nn.TransformerEncoder |
部分支持 |
功能存在限制 |
nn.TransformerDecoder |
部分支持 |
功能存在限制 |
nn.TransformerEncoderLayer |
部分支持 |
功能存在限制 |
nn.TransformerDecoderLayer |
部分支持 |
功能存在限制 |
nn.LazyConv1d |
支持 |
|
nn.LazyConv2d |
支持 |
|
nn.LazyConv3d |
支持 |
|
nn.LazyConvTranspose1d |
支持 |
|
nn.LazyConvTranspose2d |
支持 |
|
nn.LazyConvTranspose3d |
支持 |
|
nn.LazyBatchNorm1d |
支持 |
|
nn.LazyBatchNorm2d |
支持 |
|
nn.LazyBatchNorm3d |
支持 |
|
nn.LazyInstanceNorm1d |
支持 |
|
nn.LazyInstanceNorm2d |
支持 |
|
nn.LazyInstanceNorm3d |
支持 |
|
nn.LazyLinear |
支持 |
|
nn.modules.lazy.LazyModuleMixin |
支持 |
|
nn.utils.rnn.pad_sequence |
支持 |
|
nn.utils.rnn.pack_padded_sequence |
支持 |
|
nn.utils.rnn.PackedSequence |
支持 |
|
nn.utils.rnn.pad_packed_sequence |
支持 |
|
nn.utils.rnn.pack_sequence |
支持 |
|
nn.parallel.DistributedDataParallel |
部分支持 |
功能存在限制 |
nn.utils.clip_grad_norm_ |
部分支持 |
功能存在限制 |
nn.utils.clip_grad_value_ |
部分支持 |
功能存在限制 |
nn.utils.parameters_to_vector |
支持 |
|
nn.utils.vector_to_parameters |
部分支持 |
暂不支持图模式 |
nn.utils.weight_norm |
支持 |
|
nn.utils.remove_weight_norm |
支持 |
|
nn.utils.spectral_norm |
支持 |
|
nn.utils.remove_spectral_norm |
支持 |
|
nn.utils.skip_init |
支持 |
|
nn.utils.parametrizations.orthogonal |
部分支持 |
功能存在限制 |
nn.utils.parametrizations.spectral_norm |
部分支持 |
暂不支持图模式 |
nn.utils.parametrize.register_parametrization |
部分支持 |
暂不支持图模式 |
nn.utils.parametrize.remove_parametrizations |
部分支持 |
暂不支持图模式 |
nn.utils.parametrize.cached |
部分支持 |
暂不支持图模式 |
nn.utils.parametrize.is_parametrized |
部分支持 |
暂不支持图模式 |
nn.utils.parametrize.ParametrizationList |
部分支持 |
暂不支持图模式 |
nn.utils.stateless.functional_call |
部分支持 |
暂不支持图模式 |
nn.quantized.FloatFunctional |
支持 |
|
nn.quantized.FXFloatFunctional |
支持 |
|
nn.Module.add_module |
支持 |
|
nn.Module.apply |
支持 |
|
nn.Module.bfloat16 |
部分支持 |
暂不支持Ascend和GPU后端 |
nn.Module.buffers |
支持 |
|
nn.Module.children |
支持 |
|
nn.Module.double |
支持 |
|
nn.Module.eval |
支持 |
|
nn.Module.extra_repr |
支持 |
|
nn.Module.float |
支持 |
|
nn.Module.get_parameter |
支持 |
|
nn.Module.get_submodule |
支持 |
|
nn.Module.half |
支持 |
|
nn.Module.load_state_dict |
支持 |
|
nn.Module.modules |
支持 |
|
nn.Module.named_buffers |
支持 |
|
nn.Module.named_children |
支持 |
|
nn.Module.named_modules |
支持 |
|
nn.Module.named_parameters |
支持 |
|
nn.Module.register_buffer |
支持 |
|
nn.Module.register_module |
支持 |
|
nn.Module.requires_grad_ |
支持 |
|
nn.Module.register_parameter |
支持 |
|
nn.Module.state_dict |
支持 |
|
nn.Module.to |
部分支持 |
功能存在限制 |
nn.Module.to_empty |
支持 |
|
nn.Module.train |
支持 |
|
nn.Module.type |
支持 |
|
nn.Module.zero_grad |
支持 |
|
nn.Module.share_memory |
部分支持 |
功能存在限制 |
nn.Module.cpu |
支持 |
|
nn.Module.cuda |
支持 |
|
nn.Module.forward |
支持 |
|
nn.Module.get_buffer |
支持 |
|
nn.Module.get_extra_state |
支持 |
|
nn.Module.set_extra_state |
支持 |
|
nn.Module.parameters |
支持 |
|
nn.Module.register_load_state_dict_post_hook |
支持 |
|
nn.Module.register_forward_pre_hook |
支持 |
|
nn.Module.register_forward_hook |
支持 |
|
nn.Module.register_backward_hook |
部分支持 |
功能存在限制 |
nn.Module.register_full_backward_hook |
部分支持 |
功能存在限制 |
nn.modules.module.register_module_forward_pre_hook |
支持 |
|
nn.modules.module.register_module_forward_hook |
支持 |
|
nn.modules.module.register_module_backward_hook |
部分支持 |
功能存在限制 |
nn.modules.module.register_module_full_backward_hook |
部分支持 |
功能存在限制 |