python - Pytorch 张量表示具有颜色值的 3D 网格

Question

给定表示 3D 网格上 X、Y、Z 坐标密度的密度值（标量）列表，我将如何创建可以存储此信息的单个张量？

例如，一个尺寸为 1x20x20x20 的张量将表示一个 20x20x20 的网格，这样：

print(tensor[:,x1,y1,z1])
0.6

print(tensor[:,x2,y2,z2])
0.4

标签： pythonpytorchtensortorch

Answer 1

根据您的评论，您想将(2000)1D 张量转换为(1x20x20x20)4D 张量。

假设您的初始张量布局如下：

X = torch.tensor([111,112,113,121,122,123,131,132,133,211,212,213,221,222,223,231,232,233,311,312,313,321,322,323,331,332,333])

这就像在其上使用视图一样简单：

xyz = X.view(3, 3, 3, 1)

并且，X[0, 1, 2]会如愿给你[123]。