在Open3D的open3d.core.Tensor类中,eq_函数用于将当前Tensor对象的元素值与指定的数值进行比较,并在比较结果为相等时将当前Tensor对象的元素值修改为1,否则修改为0。
open3d.core.Tensor.eq_(other: float, *, out: Optional[open3d.core.Tensor] = None) -> open3d.core.Tensor
import open3d.core as o3c
a = o3c.Tensor([[1, 2], [3, 4]])
b = o3c.Tensor([[1, 3], [3, 4]])
out = o3c.Tensor()
a.eq_(2) # 修改a中等于2的元素值为1,其他元素值为0
print(a) # [[0, 1], [0, 0]]
b.eq_(3, out=out) # 将b中等于3的元素值修改为1,其他元素值为0,输出到指定Tensor对象
print(out) # [[0, 1], [1, 0]]