比较与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 |
- |
详见通用差异参数表 |