Solved by KFrank in post #2 
              
              
                Hi J! 

You may use the expression for element-wise multiplication, C = A * B, 
and pytorch will use broadcasting to multiply all of the images and channels 
in A by B. 
Note that broadcasting treats “missing” leading dimensions as if they were 
singleton dimensions, so C = A * B.unsqueeze (0).unsqu…