is there a way to broadcast tensor in tensordot operation in tensorflow?

Viewed 936

I want to multiply stacked matrix which is expressed in tensor form.

tensor.shape == [2,5,7,6]

where 2 and 5 is size of batch,

tensor2.shape == [5,6,8]

where 5 is batch size.

In numpy, tensor2 is automatically broadcasted to [2,5,7,6] tensor

so I can easily use np.matmul(tensor,tensor2)

but in tensorflow, error occurs.

I tried tf.expand_dims(tensor2,0) but this also doesn't work

is there any way to broadcast tensor in tensorflow?

2 Answers
Related