PyTorch与MindSpore API映射表
由社区提供的PyTorch APIs和MindSpore APIs之间的映射,可能在参数、输入、输出、逻辑功能和特定场景等方面存在差异,可详见各API描述或已提供的差异对比。
也欢迎更多的MindSpore开发者参与完善映射内容。
PyTorch 1.8.1
torch
PyTorch 1.8.1 APIs |
MindSpore APIs |
说明 |
---|---|---|
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
功能一致 |
||
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
功能一致 |
||
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
torch.distributed
PyTorch 1.8.1 APIs |
MindSpore APIs |
说明 |
---|---|---|
torch.nn
PyTorch 1.8.1 APIs |
MindSpore APIs |
说明 |
---|---|---|
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致,PyTorch存在弃用参数 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
torch.nn.functional
PyTorch 1.8.1 APIs |
MindSpore APIs |
说明 |
---|---|---|
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
功能一致,PyTorch存在弃用参数 |
||
功能一致,参数名不一致 |
||
功能一致,参数名不一致 |
||
torch.Tensor
PyTorch 1.8.1 APIs |
MindSpore APIs |
说明 |
---|---|---|
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致 |
||
功能一致,MindSpore未定义代表内存格式的 |
torch.optim
PyTorch 1.8.1 APIs |
MindSpore APIs |
说明 |
---|---|---|
torch.utils
PyTorch 1.8.1 APIs |
MindSpore APIs |
说明 |
---|---|---|
None |
||
PyTorch 1.5.0
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.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 |
说明 |
---|---|---|
torchtext
TorchText 0.10.0 APIs |
MindSpore APIs |
说明 |
---|---|---|
torchvision
TorchVision 0.10.0 APIs |
MindSpore APIs |
说明 |
---|---|---|