In the columns, I think that you need to add a 0 where I draw the red line, and to erase the 0 where I draw a cross.
Then maybe add "weights" and "image" or W.view(-1) @ x.view(-1) to the two different columns, and recalculate the dot product (don't know if the value is correct in the second case).
In the columns, I think that you need to add a 0 where I draw the red line, and to erase the 0 where I draw a cross.
Then maybe add "weights" and "image" or W.view(-1) @ x.view(-1) to the two different columns, and recalculate the dot product (don't know if the value is correct in the second case).
Hope that this is correct now, thanks.