def isclose(a: Tensor, b: Tensor, rtol: float = 1e-05, atol: float = 1e-08) -> Tensor:
该函数用于比较两个张量 a
和 b
中值是否接近。
a
(Tensor
):用于比较的第一个张量。b
(Tensor
):用于比较的第二个张量。rtol
(float
):相对容忍度,如果 |a - b| <= atol + rtol * |b|
,则返回 True
。atol
(float
):绝对容忍度,如果 |a - b| <= atol + rtol * |b|
,则返回 True
。(Tensor
):一个布尔型张量,元素的值为是否符合容忍度限制条件。
import open3d.core as o3c
import numpy as np
a = o3c.Tensor(np.array([1.23, 4.56]))
b = o3c.Tensor(np.array([1.23001, 4.56001]))
isclose = o3c.isclose(a, b, rtol=1e-03, atol=1e-05)
print(isclose) # 输出 [True, True]
如果a
和b
的形状不相同,则会引发 ValueError
。