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

Question

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

标签： pythontensorflowmatrixmatrix-multiplication

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