python - Pytorch 张量将带有 x 和 y 坐标的 z 值放入网格中

Question

我有一个形状为 [3,N] 的火炬张量，其中三个维度分别对应于 x、y、z 值，N 是数据点的数量。即，张量看起来像这样：

[x1, y1, z1]
[x2, y2, z2]
    ...
[xN, yN, zN]

张量的 x&y 值对应于图像中的 x&y 坐标，而 z 值对应于位置 [x,y] 处的亮度。现在我想做的是根据张量的值重建图像。

通常，我使用torch.tensor.put_（Pytorch 参考）来执行此操作。

问题是，一些 x&y 组合在图像中出现多次，即我有多个位置 [x,y] 的颜色值，我只想在每个位置保持最高 z 值。

例如，让我们拍摄 2x2 图像。然后张量可能如下所示：

[0,0,1]
[0,0,3]
[0,1,0]
[1,0,1]
[1,1,1]
[1,1,3]
[1,1,5]

然后所需的输出将是

y
^
|  [0, 5]
|  [3, 1]
|
--------> x

有谁知道我如何在不迭代所有数据点 N的情况下实现这一点，类似于该torch.tensor.put_方法？

标签： pythonpytorch