比较与torch.abs的功能差异

torch.abs

torch.abs(input, *, out=None) -> Tensor

更多内容详见torch.abs。

mindspore.ops.abs

mindspore.ops.abs(x) -> Tensor

更多内容详见mindspore.ops.abs。

差异对比

PyTorch：计算输入的绝对值。

MindSpore：MindSpore此API实现功能与PyTorch一致，仅参数名不同。

分类	子类	PyTorch	MindSpore	差异
参数	参数1	input	x	功能一致，参数名不同
	参数2	out	-	不涉及

代码示例1

两API实现功能一致，用法相同。

# PyTorch
import torch
from torch import tensor

input = torch.tensor([-1, 1, 0], dtype=torch.float32)
output = torch.abs(input).numpy()
print(output)
# [1. 1. 0.]

# MindSpore
import numpy as np
import mindspore
import mindspore.ops as ops
from mindspore import Tensor

x = Tensor(np.array([-1, 1, 0]), mindspore.float32)
output = ops.abs(x).asnumpy()
print(output)
# [1. 1. 0.]