比较与torch.floor的差异
torch.floor
torch.floor(input, *, out=None) -> Tensor
更多内容详见torch.floor。
mindspore.ops.floor
mindspore.ops.floor(input) -> Tensor
更多内容详见mindspore.ops.floor。
差异对比
MindSpore此API功能与PyTorch一致,参数支持的数据类型有差异。
PyTorch:参数 input
的dtype可以是 int
和 float
。
MindSpore:参数 input
的dtype可以是 float16
,float32
和 float64
。
分类 |
子类 |
PyTorch |
MindSpore |
差异 |
---|---|---|---|---|
参数 |
参数1 |
input |
input |
二者均为Tensor,torch.floor的参数 |
参数2 |
out |
- |
详见通用差异参数表 |