Тензорное преобразование факела с координатами в другом тензоре - PullRequest
0 голосов
/ 10 октября 2018

Я хотел бы найти способ создать тензор Факела 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]]

Способ размещения координат может быть другим, если это улучшит код.Я не хочу использовать, потому что эта операция в производстве займет много времени.

Буду признателен за помощь.

...