我有3个尺寸A = P * N，B = Q * N和C = R * N的稀疏张量 . 计算乘积矩阵A * B * C的有效方法是什么，使得乘积矩阵的维数在张量流中是P * Q * R.
我已尝试使用tf.matmul，然后使用tf.reshape，但它不会为产品矩阵提供上面指定的维度 .
谢谢 .
tf.einsum 应该帮到你 . 虽然由于尺寸与矩阵乘法不相容，我并不确定 A*B*C 的意思 . 可能是这样的：
tf.einsum
A*B*C
R = tf.einsum('il,jl,kl->ijk', A, B, C)
1 回答
tf.einsum应该帮到你 . 虽然由于尺寸与矩阵乘法不相容，我并不确定A*B*C的意思 . 可能是这样的：