Я хотел бы найти способ создать тензор Факела X, являющийся преобразованием другого тензора Y с учетом координат в матрице Z.
Например, у меня есть 2-мерная матрица 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]]
Способ размещения координат может быть другим, если это улучшит код.Я не хочу использовать, потому что эта операция в производстве займет много времени.
Буду признателен за помощь.