比较与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.Tensor.sum可以通过入参initial配置求和的起始值，其他入参两接口设定相同。

代码示例

from mindspore import Tensor
import mindspore

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

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