如何在张量流中将k 2x2矩阵乘以k 2x2矩阵？

Question

我在张量流中有2个3D张量，其中两个张量的形状为Kx2x2。张量表示一组2x2矩阵。有没有办法将第一个张量中的2x2矩阵与第二个张量中的对应矩阵相乘，以便最后得到Kx2x2张量？

Answer 1

tf.matmul是矩阵乘法的正确运算符。如果张量的尺寸超过2，则内部2指定矩阵的形状。因此，两个张量的形状必须分别为[a 1， a 2，...，a n，x，y]和[a 1， a 2，...，a n，y，z]（[K ，2，2]（在OP的情况下）。

样例代码

# Suppose X and Y are two tensors of the shape [K, 2, 2]

result = tf.matmul(X, Y)