3d次2d矩阵Tensorflow

Question

假设我有两个不同尺寸的张量。

ionic cordova run android

我想对这两个张量进行矩阵乘法，以便得到形状为<tf.Tensor 'embedding_lookup:0' shape=(?, 128) dtype=float32> <tf.Tensor 'Neg:0' shape=(?, 15, 128) dtype=float32> 或(?, 15, 1)的最终张量。我怎么能这样做？

Answer 1

分别以X和Y作为2D和3D输入，我们可以使用tf.einsum -

tf.einsum('ij,ikj->ik',X,Y)

输出的形状为(?, 15)。