我将非常感谢以下方面的帮助：
给出了两个张量
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中做到这一点？
提前致谢！
你看过tf.einsum吗？
https://www.tensorflow.org/versions/r1.0/api_docs/python/tf/einsum
tf.einsum('abd,acd->abcd', A, B) 能为您提供您想要的产品吗？
tf.einsum('abd,acd->abcd', A, B)
1 回答
你看过tf.einsum吗？
https://www.tensorflow.org/versions/r1.0/api_docs/python/tf/einsum
tf.einsum('abd,acd->abcd', A, B)能为您提供您想要的产品吗？