比较与torch.polygamma的差异

查看源文件

torch.polygamma

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

更多内容详见torch.polygamma

mindspore.ops.polygamma

mindspore.ops.polygamma(n, input) -> Tensor

更多内容详见mindspore.ops.polygamma

差异对比

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

PyTorch:参数 input 的dtype可以是 intfloat

MindSpore:参数 input 的dtype可以是 float16float32float64

分类

子类

PyTorch

MindSpore

差异

参数

参数1

n

n

-

参数2

input

input

二者均为Tensor,torch.polygamma的参数 input 的dtype可以是 intfloat ,mindspore.ops.polygamma的参数 input 的dtype可以是 float16float32float64

参数3

out

-

详见通用差异参数表