is_cuda()
是Open3D的 open3d.core.Tensor
类的一个函数,用于检查张量是否在CUDA设备上。
is_cuda()
布尔值。如果张量在CUDA设备上,则返回True
,否则返回False
。
import open3d.core as o3c
# 用numpy数组创建一个张量
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
ten = o3c.Tensor(arr)
# 检查张量是否在CUDA设备上
if ten.is_cuda():
print('张量在CUDA设备上')
else:
print('张量不在CUDA设备上')
is_cuda()
函数只能在Open3D的 open3d.core.Tensor
类的实例上调用,不能在其他类或对象上调用。