我有3个尺寸A = P * N，B = Q * N和C = R * N的稀疏张量。计算乘积矩阵A * B * C的有效方法是什么，使得乘积矩阵的维数在张量流中是P * Q * R.
我尝试使用tf.matmul，然后使用tf.reshape，但它不会给出带有上述尺寸的产品矩阵。
感谢。
答案 0 :(得分：0)
tf.einsum应该帮助你。虽然我并不确定A*B*C的含义，因为维度与矩阵乘法不相容。可能是这样的：
R = tf.einsum('il,jl,kl->ijk', A, B, C)