比较与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 |
- |