Tensor element wise multiplication, dimension is different

[rasbt (Sebastian Raschka)]

the .view(-1) approach should work fine for you then