比较与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: 参数 inputother 的dtype可以是 boolintfloat

MindSpore: 参数 x1x2 的dtype可以是 int32float32float16

分类

子类

PyTorch

MindSpore

差异

参数

参数1

input

x1

二者参数名不同。二者均为Tensor,但参数 input 的dtype可以是 boolintfloat ,参数 x1 的dtype可以是 int32float32float16

参数2

other

x2

二者参数名不同。二者均为Tensor,但参数 other 的dtype可以是 boolintfloat ,参数 x2 的dtype可以是 int32float32float16

参数3

rtol

rtol

-

参数4

atol

atol

-

参数5

equal_nan

equal_nan

-