PyTorch与MindSpore API映射表
由社区提供的PyTorch APIs和MindSpore APIs之间的映射,可能在参数、输入、输出、逻辑功能和特定场景等方面存在差异,可详见各API描述或已提供的差异对比。
也欢迎更多的MindSpore开发者参与完善映射内容。
torch
PyTorch 1.5.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
torch.autograd
PyTorch 1.5.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
torch.cuda
PyTorch 1.5.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
torch.distributed
PyTorch 1.5.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
torch.distributions
PyTorch 1.5.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
torch.nn
PyTorch 1.5.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
torch.nn.functional
PyTorch 1.5.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
torch.nn.init
PyTorch 1.5.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
torch.optim
PyTorch 1.5.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
torch.Tensor
PyTorch 1.5.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
将mindspore.ops.tensor_scatter_elements中的reduction设置为”add” |
||
torch.utils
PyTorch 1.5.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
None |
||
torchtext
TorchText 0.10.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
torchvision
TorchVision 0.10.0 APIs |
MindSpore APIs |
说明 |
---|---|---|