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

【问题标题】：how to multiply by batches in keras backend or tf如何在 keras 后端或 tf 中按批次相乘
【发布时间】：2018-06-06 20:35:05
【问题描述】：

我有两个形状的张量 (batch size,15,500) , (batch size,500,98) 我想将它们乘以每个批量大小的矩阵乘法以获得（批量大小，15,98）我怎样才能在 tensorflow 或 keras 后端做到这一点我可以使用批处理点

【问题讨论】：

它们是层还是 tensors 在自定义函数中？
自定义函数中的张量

标签： python tensorflow keras

【解决方案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 重复该操作

【讨论】：