比较与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
配置求和的起始值,其他入参两接口设定相同。
代码示例
import mindspore as ms
a = ms.Tensor([10, -5], ms.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.)