Как умножить матрицы k 2x2 на матрицы k 2x2 в тензорном потоке?

Question

У меня есть 2 3D тензора в тензорном потоке, где два тензора имеют форму Kx2x2. Тензоры представляют собой набор матриц 2x2. Есть ли способ умножить матрицы 2x2 в первом тензоре на соответствующую матрицу во втором, чтобы в конце я получил тензор Kx2x2?

Answer 1

Вы можете сделать tf.matmul (a, b).

Согласно тензорной документации,

tf.matmul возвращает: Тензор того же типа, что и a и b, где каждая самая внутренняя матрица является произведением соответствующих матриц в a и b.

output [..., i, j] = sum_k (a [..., i, k] * b [..., k, j]) для всех индексов i, j.

https://www.tensorflow.org/api_docs/python/tf/matmul

Answer 2

tf.matmul - правильный оператор для умножения матриц. Если размер тензора больше 2, то внутренние 2 задают форму матрицы. Следовательно, форма двух тензоров должна быть [a _1, a ₂ , ..., a _n , x, y] и [a _1, a ₂ , ..., a _n , y, z] соответственно ([K, 2, 2] в случае ОП).

Пример кода

# Suppose X and Y are two tensors of the shape [K, 2, 2]
result = tf.matmul(X, Y)