检查一个Open3D的 HashSet
是否在CUDA设备上分配了内存。
is_cuda() -> bool
无
返回一个布尔值,表示 HashSet
是否在CUDA设备上分配了内存。
import open3d.core as o3c
device = o3c.Device("cuda:0") # 分配一个CUDA设备
points = o3c.Tensor([[0.1, 0.2, 0.3],
[1.1, 1.2, 1.3],
[2.1, 2.2, 2.3]], device=device)
hash_set = o3c.HashSet.create_from_point_cloud(points, 0.02)
print(hash_set.is_cuda()) # True
device = o3c.Device("cpu") # 将点云转移到CPU上
points = points.to(device)
hash_set = o3c.HashSet.create_from_point_cloud(points, 0.02)
print(hash_set.is_cuda()) # False
无