比较与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.)