首页 文章

如何在keras后端或tf中按批次相乘

提问于
浏览
2

我有两个形状张量(批量大小,15,500),(批量大小,500,98)我想将它们乘以矩阵乘法获取每个批量大小(批量大小,15,98)我如何在tensorflow中做到这一点或keras后端我可以使用批量点

1 回答

  • 0

    您可以使用

    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重复

相关问题