I have two tensors. A has shape (N, C, H, W) and B has shape (C). Now I want to multiply both tensors along C.
Currently I use torch.einsum("ijkl,j->ijkl", A, B) and it seems to work.
I would like to know if there is a better or more intuitive way to do this? Mabe with .view()?
Thanks!