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()
函数检查。