比较与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可以是 int
或float
。
MindSpore:参数 input
的dtype可以是 float16
,float32
或 float64
。
分类 |
子类 |
PyTorch |
MindSpore |
差异 |
---|---|---|---|---|
参数 |
参数1 |
n |
n |
- |
参数2 |
input |
input |
二者均为Tensor,torch.polygamma的参数 |
|
参数3 |
out |
- |
详见通用差异参数表 |