is_cpu() 是 Open3D 中的核心类 Tensor 的一个成员函数,用于检查 Tensor 对象是否在 CPU 上进行操作。
is_cpu() -> bool
无。
如果 Tensor 对象在 CPU 上进行操作,则返回 True;否则返回 False。
import open3d.core as o3c
t = o3c.Tensor.zeros((3, 3), o3c.Dtype.Float32)
print(t.is_cpu()) # True
# 将张量移动到GPU上
device = o3c.Device("cuda:0")
t = t.to(device)
print(t.is_cpu()) # False
如果需要在 GPU 上进行操作,可以使用 to() 函数将 Tensor 对象移动到 GPU 上,并使用 is_cuda() 函数检查。