HI guys:

there are two multi-dimension matrix

a=[64,10,1,1]; b=[64,10,28,28]

I found that a*b can be implemented in TensorFlow. And the result is also [64,10,28,28].

But when I used Pytorch, it didn’t work.

So how should I do?

THANKS

HI guys:

there are two multi-dimension matrix

a=[64,10,1,1]; b=[64,10,28,28]

I found that a*b can be implemented in TensorFlow. And the result is also [64,10,28,28].

But when I used Pytorch, it didn’t work.

So how should I do?

THANKS

I assume that you are looking for element-wise multiplication where the last two dimensions are broadcasted for `a`

?

If so, you can do this with `a.expand_as(b) * b`