diff --git a/resource/api_mapping.md b/resource/api_mapping.md new file mode 100644 index 0000000000000000000000000000000000000000..bb3471b378ec24b365eb784563d60598c72dc3d0 --- /dev/null +++ b/resource/api_mapping.md @@ -0,0 +1,160 @@ +# API Mapping + +Mapping between PyTorch APIs and MindSpore APIs, which is provided by the community. + +| PyTorch APIs | MindSpore APIs | +|------------------------------------------------------|------------------------------------------------------------------------| +| torch.abs | mindspore.ops.operations.Abs | +| torch.acos | mindspore.ops.operations.ACos | +| torch.add | mindspore.ops.operations.TensorAdd | +| torch.argmax | mindspore.ops.operations.Argmax | +| torch.argmin | mindspore.ops.operations.Argmin | +| torch.atan2 | mindspore.ops.operations.Atan2 | +| torch.bmm | mindspore.ops.operations.BatchMatMul | +| torch.cat | mindspore.ops.operations.Concat | +| torch.chunk | mindspore.ops.operations.Split | +| torch.clamp | mindspore.ops.composite.clip_by_value | +| torch.cos | mindspore.ops.operations.Cos | +| torch.cuda.device_count | mindspore.communication.get_group_size | +| torch.cuda.set_device | mindspore.context.set_context | +| torch.cumprod | mindspore.ops.operations.CumProd | +| torch.cumsum | mindspore.ops.operations.CumSum | +| torch.distributed.all_gather | mindspore.ops.operations.AllGather | +| torch.distributed.all_reduce | mindspore.ops.operations.AllReduce | +| torch.distributed.get_rank | mindspore.communication.get_rank | +| torch.distributed.init_process_group | mindspore.communication.init | +| torch.div | mindspore.ops.operations.Div | +| torch.eq | mindspore.ops.operations.Equal | +| torch.erfc | mindspore.ops.operations.Erfc | +| torch.exp | mindspore.ops.operations.Exp | +| torch.eye | mindspore.ops.operations.Eye | +| torch.flatten | mindspore.ops.operations.Flatten | +| torch.floor | mindspore.ops.operations.Floor | +| torch.load | mindspore.train.serialization.load_checkpoint | +| torch.log | mindspore.ops.operations.Log | +| torch.log1p | mindspore.ops.operations.Log1p | +| torch.matmul | mindspore.ops.operations.MatMul | +| torch.max | mindspore.ops.operations.Maximum | +| torch.mean | mindspore.ops.operations.ReduceMean | +| torch.min | mindspore.ops.operations.Minimum | +| torch.mm | mindspore.ops.operations.MatMul | +| torch.mul | mindspore.ops.operations.Mul | +| torch.nn.AdaptiveAvgPool2d | mindspore.ops.operations.ReduceMean | +| torch.nn.AvgPool1d | mindspore.nn.AvgPool1d | +| torch.nn.AvgPool2d | mindspore.nn.AvgPool2d | +| torch.nn.BatchNorm1d | mindspore.nn.BatchNorm1d | +| torch.nn.BatchNorm2d | mindspore.nn.BatchNorm2d | +| torch.nn.Conv2d | mindspore.nn.Conv2d | +| torch.nn.ConvTranspose2d | mindspore.nn.Conv2dTranspose | +| torch.nn.CrossEntropyLoss | mindspore.nn.SoftmaxCrossEntropyWithLogits | +| torch.nn.CTCLoss | mindspore.ops.operations.CTCLoss | +| torch.nn.Dropout | mindspore.nn.Dropout | +| torch.nn.Embedding | mindspore.nn.Embedding | +| torch.nn.Flatten | mindspore.nn.Flatten | +| torch.nn.functional.adaptive_avg_pool2d | mindspore.nn.AvgPool2d | +| torch.nn.functional.avg_pool2d | mindspore.ops.operations.AvgPool | +| torch.nn.functional.binary_cross_entropy | mindspore.ops.operations.BinaryCrossEntropy | +| torch.nn.functional.conv2d | mindspore.ops.operations.Conv2D | +| torch.nn.functional.elu | mindspore.ops.operations.Elu | +| torch.nn.functional.interpolate | mindspore.ops.operations.ResizeBilinear | +| torch.nn.functional.log_softmax | mindspore.nn.LogSoftmax | +| torch.nn.functional.normalize | mindspore.ops.operations.L2Normalize | +| torch.nn.functional.one_hot | mindspore.ops.operations.OneHot | +| torch.nn.functional.pad | mindspore.ops.operations.Pad | +| torch.nn.functional.pixel_shuffle | mindspore.ops.operations.DepthToSpace | +| torch.nn.functional.relu | mindspore.ops.operations.ReLU | +| torch.nn.functional.softmax | mindspore.ops.operations.Softmax | +| torch.nn.functional.softplus | mindspore.ops.operations.Softplus | +| torch.nn.GroupNorm | mindspore.nn.GroupNorm | +| torch.nn.init.constant_ | mindspore.common.initializer.Constant | +| torch.nn.init.uniform_ | mindspore.common.initializer.Uniform | +| torch.nn.L1Loss | mindspore.nn.L1Loss | +| torch.nn.LayerNorm | mindspore.nn.LayerNorm | +| torch.nn.LeakyReLU | mindspore.nn.LeakyReLU | +| torch.nn.Linear | mindspore.nn.Dense | +| torch.nn.LSTM | mindspore.nn.LSTM | +| torch.nn.LSTMCell | mindspore.nn.LSTMCell | +| torch.nn.MaxPool2d | mindspore.nn.MaxPool2d | +| torch.nn.Module | mindspore.nn.Cell | +| torch.nn.Module.load_state_dict | mindspore.train.serialization.load_param_into_net | +| torch.nn.ModuleList | mindspore.nn.CellList | +| torch.nn.MSELoss | mindspore.nn.MSELoss | +| torch.nn.Parameter | mindspore.Parameter | +| torch.nn.ParameterList | mindspore.ParameterTuple | +| torch.nn.PixelShuffle | mindspore.ops.operations.DepthToSpace | +| torch.nn.PReLU | mindspore.nn.PReLU | +| torch.nn.ReLU | mindspore.nn.ReLU | +| torch.nn.ReplicationPad2d | mindspore.nn.Pad | +| torch.nn.Sequential | mindspore.nn.SequentialCell | +| torch.nn.Sigmoid | mindspore.nn.Sigmoid | +| torch.nn.SmoothL1Loss | mindspore.nn.SmoothL1Loss | +| torch.nn.Softmax | mindspore.nn.Softmax | +| torch.nn.Tanh | mindspore.nn.Tanh | +| torch.nn.Unfold | mindspore.nn.Unfold | +| torch.nn.Upsample | mindspore.ops.operations.ResizeBilinear | +| torch.norm | mindspore.nn.Norm | +| torch.numel | mindspore.ops.operations.Size | +| torch.ones | mindspore.ops.operations.OnesLike | +| torch.ones_like | mindspore.ops.operations.OnesLike | +| torch.optim.Adam | mindspore.nn.Adam | +| torch.optim.AdamW | mindspore.nn.AdamWeightDecay | +| torch.optim.lr_scheduler.CosineAnnealingWarmRestarts | mindspore.nn.dynamic_lr.cosine_decay_lr | +| torch.optim.lr_scheduler.StepLR | mindspore.nn.dynamic_lr.piecewise_constant_lr | +| torch.optim.Optimizer.step | mindspore.nn.TrainOneStepCell | +| torch.optim.RMSprop | mindspore.nn.RMSProp | +| torch.optim.SGD | mindspore.nn.SGD | +| torch.pow | mindspore.ops.operations.Pow | +| torch.prod | mindspore.ops.operations.ReduceProd | +| torch.randn | mindspore.ops.operations.TruncatedNormal | +| torch.round | mindspore.ops.operations.Round | +| torch.save | mindspore.train.serialization.save_checkpoint | +| torch.sigmoid | mindspore.ops.operations.Sigmoid | +| torch.sin | mindspore.ops.operations.Sin | +| torch.sparse.FloatTensor | mindspore.Tensor | +| torch.split | mindspore.ops.operations.Split | +| torch.sqrt | mindspore.ops.operations.Sqrt | +| torch.squeeze | mindspore.ops.operations.Squeeze | +| torch.stack | mindspore.ops.operations.Pack | +| torch.std_mean | mindspore.ops.operations.ReduceMean | +| torch.sum | mindspore.ops.operations.ReduceSum | +| torch.tanh | mindspore.ops.operations.Tanh | +| torch.tensor | mindspore.Tensor | +| torch.Tensor | mindspore.Tensor | +| torch.Tensor.chunk | mindspore.ops.operations.Split | +| torch.Tensor.fill_ | mindspore.ops.operations.Fill | +| torch.Tensor.float | mindspore.ops.operations.Cast | +| torch.Tensor.mm | mindspore.ops.operations.MatMul | +| torch.Tensor.mul | mindspore.ops.operations.Mul | +| torch.Tensor.pow | mindspore.ops.operations.Pow | +| torch.Tensor.repeat | mindspore.ops.operations.Tile | +| torch.Tensor.requires_grad_ | mindspore.Parameter.requires_grad | +| torch.Tensor.round | mindspore.ops.operations.Round | +| torch.Tensor.scatter | mindspore.ops.operations.ScatterNd | +| torch.Tensor.sigmoid | mindspore.nn.Sigmoid | +| torch.Tensor.sign | mindspore.ops.operations.Sign | +| torch.Tensor.size | mindspore.ops.operations.Shape | +| torch.Tensor.sqrt | mindspore.ops.operations.Sqrt | +| torch.Tensor.sub | mindspore.ops.operations.Sub | +| torch.Tensor.t | mindspore.ops.operations.Transpose | +| torch.Tensor.transpose | mindspore.ops.operations.Transpose | +| torch.Tensor.unsqueeze | mindspore.ops.operations.ExpandDims | +| torch.Tensor.view | mindspore.ops.operations.Reshape | +| torch.Tensor.zero_ | mindspore.ops.operations.ZerosLike | +| torch.transpose | mindspore.ops.operations.Transpose | +| torch.unbind | mindspore.ops.operations.Unpack | +| torch.unsqueeze | mindspore.ops.operations.ExpandDims | +| torch.utils.data.DataLoader | mindspore.DatasetHelper | +| torch.utils.data.Dataset | mindspore.dataset.MindDataset | +| torch.utils.data.distributed.DistributedSampler | mindspore.dataset.DistributedSampler | +| torch.zeros | mindspore.ops.operations.ZerosLike | +| torch.zeros_like | mindspore.ops.operations.ZerosLike | +| torchvision.datasets.ImageFolder | mindspore.dataset.ImageFolderDatasetV2 | +| torchvision.ops.nms | mindspore.ops.operations.NMSWithMask | +| torchvision.ops.roi_align | mindspore.ops.operations.ROIAlign | +| torchvision.transforms.CenterCrop | mindspore.dataset.transforms.vision.py_transforms.CenterCrop | +| torchvision.transforms.ColorJitter | mindspore.dataset.transforms.vision.py_transforms.RandomColorAdjust | +| torchvision.transforms.Compose | mindspore.dataset.transforms.vision.py_transforms.ComposeOp | +| torchvision.transforms.Normalize | mindspore.dataset.transforms.vision.py_transforms.Normalize | +| torchvision.transforms.RandomHorizontalFlip | mindspore.dataset.transforms.vision.py_transforms.RandomHorizontalFlip | +| torchvision.transforms.Resize | mindspore.dataset.transforms.vision.py_transforms.Resize | +| torchvision.transforms.ToTensor | mindspore.dataset.transforms.vision.py_transforms.ToTensor | \ No newline at end of file