我有两个形状张量(批量大小,15,500),(批量大小,500,98)我想将它们乘以矩阵乘法获取每个批量大小(批量大小,15,98)我如何在tensorflow中做到这一点或keras后端我可以使用批量点
您可以使用
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重复
1 回答
您可以使用
这表示索引j的合约,这是单个矩阵乘法的样子,并对索引i重复