比较与torch.cos的差异

查看源文件

torch.cos

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

更多内容详见torch.cos。

mindspore.ops.cos

mindspore.ops.cos(input) -> Tensor

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

差异对比

MindSpore此API功能与PyTorch一致,参数支持的数据类型有差异。

PyTorch:参数 input 的dtype可以是 complex,int 或 float 。

MindSpore:参数 input 的dtype可以是 float16 ,float32 ,float64 ,complex64 或 complex128 。

分类

子类

PyTorch

MindSpore

差异

参数

参数1

input

input

二者均为Tensor,torch.cos的参数 input 的dtype可以是 complex,int 或 float ;mindspore.ops.cos的参数 input 的dtype可以是 float16 ,float32 ,float64 ,complex64 或 complex128 。

参数2

out

-

详见通用差异参数表