在张量流中乘以张量链的有效方法是什么
What is the efficient way of multiplying chain of tensors in tensorflow
我有 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 是什么意思，因为尺寸与矩阵乘法不兼容。大概是这样的：
R = tf.einsum('il,jl,kl->ijk', A, B, C)
我有 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 是什么意思，因为尺寸与矩阵乘法不兼容。大概是这样的：
R = tf.einsum('il,jl,kl->ijk', A, B, C)