Hello,
I have a 3-D weight tensor W with dimensionality (C,H,H) and 2-D input x (C,H).
My goal is to calculate a sum.  \sum_i W_i * x_i and have an output (C,H).
How I can implement it in pytorch? Could not make it work with torch.mul