比较与torch.Tensor.sum的差异

查看源文件

torch.Tensor.sum

torch.Tensor.sum(dim=None, keepdim=False, dtype=None)

更多内容详见torch.Tensor.sum

mindspore.Tensor.sum

mindspore.Tensor.sum(axis=None, dtype=None, keepdims=False, initial=None)

更多内容详见mindspore.Tensor.sum

差异对比

MindSpore此API功能与PyTorch一致,参数的个数和顺序不一致。

PyTorch:无参数 initial 。参数 keepdimdtype 的相对顺序与MindSpore不同。

MindSpore:可以通过参数 initial 配置求和的起始值。参数 keepdimdtype 的相对顺序与PyTorch不同。

分类

子类

PyTorch

MindSpore

差异

参数

参数1

dim

axis

二者参数名不同,均表示求和的指定维度

参数2

keepdim

dtype

二者参数 keepdimdtype 的相对顺序不同

参数3

dtype

keepdims

二者参数 keepdimsdtype 的相对顺序不同

参数4

-

initial

MindSpore可以通过参数 initial 配置求和的起始值,PyTorch无参数 initial

代码示例

# PyTorch
import torch

b = torch.Tensor([10, -5])
print(torch.Tensor.sum(b))
# tensor(5.)

# MindSpore
import mindspore as ms

a = ms.Tensor([10, -5], ms.float32)
print(a.sum())
# 5.0
print(a.sum(initial=2))
# 7.0