open3d.core.Tensor
类是 Open3D 的核心张量类。本文档介绍 open3d.core.Tensor
的 CPU 实现。
open3d.core.Tensor
类包含了支持 CPU 的实现。以下是 open3d.core.Tensor
的 CPU 相关的 API 列表:
to(cpu_device: Union[int, str]) -> open3d.core.Tensor
:将张量数据拷贝到指定的 CPU 设备。device()
:获取张量所在的设备。下面的代码演示如何利用 open3d.core.Tensor
的 CPU 实现:
import open3d.core as o3c
# 为了演示,这里的数据是随机生成的
data = [
[1.1, 2.2, 3.3, 4.4],
[5.5, 6.6, 7.7, 8.8],
[9.9, 10.10, 11.11, 12.12]
]
# 创建张量
tensor = o3c.Tensor(data)
# 获取张量所在的设备
print(tensor.device()) # 输出 'CPU:0'
# 将张量数据拷贝到指定的 CPU 设备
cpu_tensor = tensor.to(0)
# 再次获取张量所在的设备
print(cpu_tensor.device()) # 输出 'CPU:0'