表1 torch.nn
序号	API名称	是否支持
1	Parameter	是
2	UninitializedParameter	是

表2 Containers
序号	API名称	是否支持
1	Module	是
2	ModuleDict	是
3	ModuleList	是
4	ParameterDict	是
5	ParameterList	是
6	register_module_backward_hook	是
7	register_module_forward_hook	是
8	register_module_forward_pre_hook	是
9	Sequential	是

表7 Normalization Layers
序号	API名称	是否支持
1	nn.AdaptiveLogSoftmaxWithLoss	否
2	nn.LogSoftmax	是
3	nn.Softmax	是
4	nn.Softmax2d	是
5	nn.Softmin	是

**表10** Transformer Layers
序号	API名称	是否支持
1	nn.Transformer	是
2	nn.TransformerDecoder	是
3	nn.TransformerDecoderLayer	是
4	nn.TransformerEncoder	是
5	nn.TransformerEncoderLayer	是

**表13** Sparse Layers
序号	API名称	是否支持
1	nn.Embedding	是
2	nn.EmbeddingBag	是

**表14** Distance Functions
序号	API名称	是否支持
1	nn.CosineSimilarity	是
2	nn.PairwiseDistance	是

**表15** Loss Functions
序号	API名称	是否支持
1	nn.BCELoss	是
2	nn.BCEWithLogitsLoss	是
3	nn.CosineEmbeddingLoss	是
4	nn.CrossEntropyLoss	是
5	nn.CTCLoss	是
6	nn.GaussianNLLLoss	是
7	nn.HingeEmbeddingLoss	是
8	nn.KLDivLoss	是
9	nn.L1Loss	是
10	nn.MarginRankingLoss	是
11	nn.MSELoss	是
12	nn.MultiLabelMarginLoss	是
13	nn.MultiLabelSoftMarginLoss	是
14	nn.MultiMarginLoss	否
15	nn.NLLLoss	是
16	nn.PoissonNLLLoss	是
17	nn.SmoothL1Loss	是
18	nn.SoftMarginLoss	是
19	nn.TripletMarginLoss	是
20	nn.TripletMarginWithDistanceLoss	是

**表17** Shuffle Layers
序号	API名称	是否支持
1	nn.ChannelShuffle	是

**表18** DataParallel Layers
序号	API名称	是否支持
1	nn.DataParallel	否
2	nn.parallel.DistributedDataParallel	是

**表19** Utilities
序号	API名称	是否支持
1	clip_grad_norm_	是
2	clip_grad_value_	是
3	nn.Flatten	是
4	nn.Unflatten	是
5	nn.utils.rnn.pack_padded_sequence	是
6	nn.utils.rnn.pack_sequence	是
7	nn.utils.rnn.PackedSequence	是
8	nn.utils.rnn.pad_packed_sequence	是
9	nn.utils.rnn.pad_sequence	是
10	parameters_to_vector	是
11	prune.BasePruningMethod	否
12	prune.custom_from_mask	是
13	prune.CustomFromMask	是
14	prune.global_unstructured	是
15	prune.Identity	是
16	prune.identity	是
17	prune.is_pruned	是
18	prune.l1_unstructured	是
19	prune.L1Unstructured	是
20	prune.ln_structured	是
21	prune.LnStructured	是
22	prune.PruningContainer	是
23	prune.random_structured	是
24	prune.random_unstructured	是
25	prune.RandomStructured	是
26	prune.RandomUnstructured	是
27	prune.remove	是
28	remove_spectral_norm	是
29	remove_weight_norm	是
30	spectral_norm	是
31	vector_to_parameters	是
32	weight_norm	是

**表20** Lazy Modules Initialization
序号	API名称	是否支持
1	nn.modules.lazy.LazyModuleMixin	是