函数concatenate
实现了在指定维度上连接两个指定的张量的功能。它是Open3D中open3d.core
模块的一部分。
concatenate(tensors, dim=0)
tensors
:张量列表,要连接的张量序列。dim
:一个指定连接维度的整数。默认值为0
。返回连接后的张量。
import open3d.core as o3c
import torch
x = torch.tensor([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]], dtype=torch.float32)
y = torch.tensor([[[13, 14, 15], [16, 17, 18]], [[19, 20, 21], [22, 23, 24]]], dtype=torch.float32)
o3c_x = o3c.Tensor.from_numpy(x)
o3c_y = o3c.Tensor.from_numpy(y)
o3c_res = o3c.concatenate([o3c_x, o3c_y], dim=0)
print(o3c_res.cpu().numpy())
运行结果为:
[[[ 1. 2. 3.]
[ 4. 5. 6.]]
[[ 7. 8. 9.]
[10. 11. 12.]]
[[13. 14. 15.]
[16. 17. 18.]]
[[19. 20. 21.]
[22. 23. 24.]]]
以上示例演示了如何在dim=0
维上连接两个张量。
ValueError
:如果两个输入张量的维度数不相等,将引发此异常。