Hello Mr. Knight!



 nowyouseeme:

I want to do element wise multiplication of B with A, such that B is multiplied with all 128 columns of tensor A…
I want to broadcast the element wise multiplication along dimension=1.


Give B a dimension of size 1 using unsqueeze() so that it has a
dimension from which to broadcast:
B.unsqueeze (1) * A

Best.
K. Frank