1

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

我已经尝试使用 tf.matmul 并随后使用 tf.reshape 但它不会给出具有上述尺寸的产品矩阵。

谢谢。

4

1 回答 1

0

tf.einsum应该能帮到你。虽然我不太确定你的意思，A*B*C因为维度与矩阵乘法不兼容。大概是这样的：

R = tf.einsum('il,jl,kl->ijk', A, B, C)
于 2017-02-01T23:43:19.367 回答