我想乘以张量形式表示的堆叠矩阵。
tensor.shape == [2,5,7,6]
其中 2 和 5 是批量大小，
tensor2.shape == [5,6,8]
其中 5 是批量大小。
在numpy中，tensor2自动广播到[2,5,7,6]张量
所以我可以很容易地使用 np.matmul(tensor,tensor2)
但是在tensorflow中，会出现错误。
我尝试了 tf.expand_dims(tensor2,0) 但这也不起作用
有什么办法可以在tensorflow中广播tensor吗？
请您参考如下方法:
你可以使用tf.einsum:
tf.einsum('abij,bjk->abik', tensor, tensor2)
例子:
import tensorflow as tf
x = tf.zeros((2, 5, 7, 6))
y = tf.zeros((5, 6, 8))
z = tf.einsum('abij,bjk->abik', x, y)
z.shape.as_list()
# returns [2, 5, 7, 8]
