如何在Tensorflow中进行矩阵乘法运算？

Question

我将非常感谢以下方面的帮助：

给出两个张量

A = bsz x a_len x dim

B = bsz x b_len x dim

我想做一个矩阵方式的逐元素乘法，使得长度为dim的每个向量乘以B的每个向量（dim长度）

输出应为：

bsz x a_len x b_len x dim

如何在Tensorflow中执行此操作？

提前致谢！

Answer 1

你看过tf.einsum吗？

https://www.tensorflow.org/versions/r1.0/api_docs/python/tf/einsum

tf.einsum('abd,acd->abcd', A, B)会为您提供您正在寻找的内容吗？