python - Tensorflow 中是否有一个函数可以做以下数学运算？

Question

我有两个张量，x和y，形状为[B, D]。我想做类似下面的代码

B, D = x.shape
x = tf.expand_dims(x, 1)  # [B, 1, D]
y = tf.expand_dims(y, -1) # [B, D, 1]
z = x * y                 # [B, D, D]
z = tf.reshape(z, (B, D**2))

Tensorflow 中是否有一个功能已经做到了这一点？

标签： pythontensorflow