比较与torch.log2的差异

查看源文件

torch.log2

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

更多内容详见torch.log2。

mindspore.ops.log2

mindspore.ops.log2(input) -> Tensor

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

差异对比

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

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

MindSpore:在CPU平台上,参数 input 的dtype可以是 float16 , float32 或 float64 ;在Ascend平台上,参数 input 的dtype可以是 float16 或 float32 。

分类

子类

PyTorch

MindSpore

差异

参数

参数1

input

input

二者均为Tensor,torch.log2的参数 input 的dtype可以是 int 或 float ,mindspore.ops.log2的参数 input 的dtype在CPU平台上可以是 float16 , float32 或 float64 ,在Ascend平台上可以是 float16 或 float32 。

参数2

out

-

详见通用差异参数表