在另一个张量中具有坐标的火炬张量变换

Question

我想找到一种创建Torch张量X的方法，它是给定矩阵Z中坐标的另一个张量Y的变换。

例如，我有二维矩阵X

    [[1, 2, 3],
X =  [1, 4, 2],
     [1, 2, 5]]

坐标为矩阵Z

    [[[0, 1], [1, 1], [0, 1]],
Z =  [[1, 2], [2, 0], [1, 2]],
     [[2, 1], [2, 2], [1, 2]]]

其中每个最后一个维度矢量代表矩阵X的坐标

所以我想接收矩阵Y，将其转换为X，坐标为Z。在这种情况下，Y看起来像这样：

    [[2, 4, 2],
Y =  [2, 1, 2],
     [2, 5, 2]]

如果可以改善代码，则放置坐标的方式可能会有所不同。我不想使用，因为此操作在生产中会花费很多时间。

感谢您的帮助。