如何乘以keras后端或tf的批次

Question

我有两个形状张量（批量大小，15,500），（批量大小，500,98） 我希望将它们乘以矩阵乘法，以获得每个批量大小（批量大小，15,98） 我怎么能在tensorflow或keras后端做到这一点 我可以使用批量点

Answer 1

您可以使用

A = K.placeholder((None, 15, 500))
B = K.placeholder((None, 500, 98))

C = tf.einsum("ijk,ljn->ijn", A, B)

In [84]: C
Out[84]: <tf.Tensor 'einsum/transpose_2:0' shape=(?, 15, 98) dtype=float32>

这表示索引j的合约，这是单个矩阵乘法的样子，并重复索引i