比较与torch.atan的功能差异

torch.atan

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

更多内容详见torch.atan。

mindspore.ops.atan

mindspore.ops.atan(x) -> Tensor

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

差异对比

PyTorch：逐元素计算输入Tensor的反正切值。

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

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

代码示例1

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

# PyTorch
import numpy as np
import torch
from torch import tensor

input = torch.tensor(np.array([0.2341, 1.0, 0.0, -0.6448]), dtype=torch.float32)
output = torch.atan(input).numpy()
print(output)
# [ 0.22995889  0.7853982   0.         -0.572711  ]

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

x = Tensor(np.array([0.2341, 1.0, 0.0, -0.6448]), mindspore.float32)
output = ops.atan(x)
print(output)
# [ 0.22995889  0.7853982   0.         -0.572711  ]