在张量流中乘以张量链的有效方法是什么？

Question

我有3个尺寸A = P * N，B = Q * N和C = R * N的稀疏张量。计算乘积矩阵A * B * C的有效方法是什么，使得乘积矩阵的维数在张量流中是P * Q * R.

我尝试使用tf.matmul，然后使用tf.reshape，但它不会给出带有上述尺寸的产品矩阵。

感谢。

Answer 1

tf.einsum应该帮助你。虽然我并不确定A*B*C的含义，因为维度与矩阵乘法不相容。可能是这样的：

R = tf.einsum('il,jl,kl->ijk', A, B, C)