blob: 04b62cfe375d5fd60ea9651d81af0a9234acbda3 [file] [view]
# Python / C++ API parity tracker
## torch.nn
API | Implementation Parity | Doc Parity
------------- | ------------- | -------------
torch.nn.Sequential|No|No
torch.nn.ModuleList|No|No
torch.nn.ModuleDict|No|No
torch.nn.ParameterList|No|No
torch.nn.ParameterDict|No|No
torch.nn.Conv1d|No|No
torch.nn.Conv2d|No|No
torch.nn.Conv3d|No|No
torch.nn.ConvTranspose1d|No|No
torch.nn.ConvTranspose2d|No|No
torch.nn.ConvTranspose3d|No|No
torch.nn.Unfold|No|No
torch.nn.Fold|No|No
torch.nn.MaxPool1d|No|No
torch.nn.MaxPool2d|No|No
torch.nn.MaxPool3d|No|No
torch.nn.MaxUnpool1d|No|No
torch.nn.MaxUnpool2d|No|No
torch.nn.MaxUnpool3d|No|No
torch.nn.AvgPool1d|No|No
torch.nn.AvgPool2d|No|No
torch.nn.AvgPool3d|No|No
torch.nn.FractionalMaxPool2d|No|No
torch.nn.LPPool1d|No|No
torch.nn.LPPool2d|No|No
torch.nn.AdaptiveMaxPool1d|No|No
torch.nn.AdaptiveMaxPool2d|No|No
torch.nn.AdaptiveMaxPool3d|No|No
torch.nn.AdaptiveAvgPool1d|No|No
torch.nn.AdaptiveAvgPool2d|No|No
torch.nn.AdaptiveAvgPool3d|No|No
torch.nn.ReflectionPad1d|No|No
torch.nn.ReflectionPad2d|No|No
torch.nn.ReplicationPad1d|No|No
torch.nn.ReplicationPad2d|No|No
torch.nn.ReplicationPad3d|No|No
torch.nn.ZeroPad2d|No|No
torch.nn.ConstantPad1d|No|No
torch.nn.ConstantPad2d|No|No
torch.nn.ConstantPad3d|No|No
torch.nn.ELU|No|No
torch.nn.Hardshrink|No|No
torch.nn.Hardtanh|No|No
torch.nn.LeakyReLU|No|No
torch.nn.LogSigmoid|No|No
torch.nn.MultiheadAttention|No|No
torch.nn.PReLU|No|No
torch.nn.ReLU|No|No
torch.nn.ReLU6|No|No
torch.nn.RReLU|No|No
torch.nn.SELU|No|No
torch.nn.CELU|No|No
torch.nn.Sigmoid|No|No
torch.nn.Softplus|No|No
torch.nn.Softshrink|No|No
torch.nn.Softsign|No|No
torch.nn.Tanh|No|No
torch.nn.Tanhshrink|No|No
torch.nn.Threshold|No|No
torch.nn.Softmin|No|No
torch.nn.Softmax|No|No
torch.nn.Softmax2d|No|No
torch.nn.LogSoftmax|No|No
torch.nn.AdaptiveLogSoftmaxWithLoss|No|No
torch.nn.BatchNorm1d|No|No
torch.nn.BatchNorm2d|No|No
torch.nn.BatchNorm3d|No|No
torch.nn.GroupNorm|No|No
torch.nn.SyncBatchNorm|No|No
torch.nn.InstanceNorm1d|No|No
torch.nn.InstanceNorm2d|No|No
torch.nn.InstanceNorm3d|No|No
torch.nn.LayerNorm|No|No
torch.nn.LocalResponseNorm|No|No
torch.nn.RNN|No|No
torch.nn.LSTM|No|No
torch.nn.GRU|No|No
torch.nn.RNNCell|No|No
torch.nn.LSTMCell|No|No
torch.nn.GRUCell|No|No
torch.nn.Transformer|No|No
torch.nn.TransformerEncoder|No|No
torch.nn.TransformerDecoder|No|No
torch.nn.TransformerEncoderLayer|No|No
torch.nn.TransformerDecoderLayer|No|No
torch.nn.Identity|No|No
torch.nn.Linear|No|No
torch.nn.Bilinear|No|No
torch.nn.Dropout|No|No
torch.nn.Dropout2d|No|No
torch.nn.Dropout3d|No|No
torch.nn.AlphaDropout|No|No
torch.nn.Embedding|No|No
torch.nn.EmbeddingBag|No|No
torch.nn.CosineSimilarity|No|No
torch.nn.PairwiseDistance|No|No
torch.nn.L1Loss|No|No
torch.nn.MSELoss|No|No
torch.nn.CrossEntropyLoss|No|No
torch.nn.CTCLoss|No|No
torch.nn.NLLLoss|No|No
torch.nn.PoissonNLLLoss|No|No
torch.nn.KLDivLoss|No|No
torch.nn.BCELoss|No|No
torch.nn.BCEWithLogitsLoss|No|No
torch.nn.MarginRankingLoss|No|No
torch.nn.HingeEmbeddingLoss|No|No
torch.nn.MultiLabelMarginLoss|No|No
torch.nn.SmoothL1Loss|No|No
torch.nn.SoftMarginLoss|No|No
torch.nn.MultiLabelSoftMarginLoss|No|No
torch.nn.CosineEmbeddingLoss|No|No
torch.nn.MultiMarginLoss|No|No
torch.nn.TripletMarginLoss|No|No
torch.nn.PixelShuffle|No|No
torch.nn.Upsample|No|No
torch.nn.UpsamplingNearest2d|No|No
torch.nn.UpsamplingBilinear2d|No|No
torch.nn.DataParallel|No|No
torch.nn.parallel.DistributedDataParallel|No|No
torch.nn.utils.clip_grad_norm_|No|No
torch.nn.utils.clip_grad_value_|No|No
torch.nn.utils.parameters_to_vector|No|No
torch.nn.utils.vector_to_parameters|No|No
torch.nn.utils.weight_norm|No|No
torch.nn.utils.remove_weight_norm|No|No
torch.nn.utils.spectral_norm|No|No
torch.nn.utils.remove_spectral_norm|No|No
torch.nn.utils.rnn.PackedSequence|No|No
torch.nn.utils.rnn.pack_padded_sequence|No|No
torch.nn.utils.rnn.pad_packed_sequence|No|No
torch.nn.utils.rnn.pad_sequence|No|No
torch.nn.utils.rnn.pack_sequence|No|No
torch.nn.Flatten|No|No
torch.nn.CrossMapLRN2d|No|No
torch.nn.FractionalMaxPool3d|No|No
torch.nn.GLU|No|No
torch.nn.SampleModule|Yes|Yes