比较与torch.isclose的差异
torch.isclose
torch.isclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False) -> Tensor
更多内容详见torch.isclose。
mindspore.ops.isclose
mindspore.ops.isclose(x1, x2, rtol=1e-05, atol=1e-08, equal_nan=False) -> Tensor
更多内容详见mindspore.ops.isclose。
差异对比
MindSpore此API功能与PyTorch一致,参数支持的数据类型有差异。
PyTorch: 参数 input
和 other
的dtype可以是 bool
,int
和 float
。
MindSpore: 参数 x1
和 x2
的dtype可以是 int32
, float32
和 float16
。
分类 |
子类 |
PyTorch |
MindSpore |
差异 |
---|---|---|---|---|
参数 |
参数1 |
input |
x1 |
二者参数名不同。二者均为Tensor,但参数 |
参数2 |
other |
x2 |
二者参数名不同。二者均为Tensor,但参数 |
|
参数3 |
rtol |
rtol |
- |
|
参数4 |
atol |
atol |
- |
|
参数5 |
equal_nan |
equal_nan |
- |