比较与torch.cos的功能差异

torch.cos

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

更多内容详见torch.cos。

mindspore.ops.cos

mindspore.ops.cos(x) -> Tensor

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

差异对比

PyTorch：返回一个新的张量，内容为输入元素的余弦。

MindSpore：功能一致，仅参数名不同。

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

代码示例

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

# 样例1 -> 代码示例1
# PyTorch
import torch

a = torch.tensor([0.24, 0.83, 0.31, 0.09],dtype=torch.float32)
output = torch.cos(a)
output = output.detach().numpy()
print(output)
# [0.971338   0.6748758  0.95233357 0.9959527 ]

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

x = Tensor(np.array([0.24, 0.83, 0.31, 0.09]), mindspore.float32)
output = ops.cos(x)
print(output)
# [0.971338 0.6748758 0.95233357 0.9959527 ]