在Open3D中，open3d.core.Tensor.flatten()方法可以将一个多维的张量(也称为数组)变成一个一维的张量。
open3d.core.Tensor.flatten(start_dim=0, end_dim=-1)
start_dim (int, 可选): 压缩开始的维度。默认为0。end_dim (int, 可选): 压缩结束的维度。默认为-1。返回一个一维的张量。
import open3d.core as o3c
import numpy as np
a = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
tensor_a = o3c.Tensor(a)
# 压缩所有维度
b = tensor_a.flatten()
print(b) # 输出一个长度为12的张量
# 压缩除第一维以外的所有维度
c = tensor_a.flatten(start_dim=1)
print(c) # 输出一个长度为12的张量
# 压缩第二维和第三维
d = tensor_a.flatten(start_dim=1, end_dim=2)
print(d) # 输出一个长度为6的张量
如果 start_dim 或 end_dim 不是整数，则会引发 TypeError。如果 start_dim 大于等于 tensor_a 的维数，则会引发 ValueError。如果 end_dim 小于等于 start_dim 或大于等于 tensor_a 的维数，则同样会引发 ValueError。