3d次2d矩阵Tensorflow

问题描述 投票:0回答:1

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

<tf.Tensor 'embedding_lookup:0' shape=(?, 128) dtype=float32>
<tf.Tensor 'Neg:0' shape=(?, 15, 128) dtype=float32>

我想对这两个张量进行矩阵乘法,这样我得到一个形状为(?, 15, 1)(?, 15)的最终张量。我怎么能这样做?

python matrix tensorflow
1个回答
3
投票

XY分别作为2D3D输入,我们可以使用tf.einsum -

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

输出的形状为(?, 15)

© www.soinside.com 2019 - 2024. All rights reserved.