If I don’t unsqueeze it but use the input as is

[262144,1]

Then I get this error:

RuntimeError: Expected 3-dimensional input for 3-dimensional weight [1024, 1, 4], but got 1-dimensional input of size [1] instead

I actually have [262144,1] tensor, how come it sees 1 dim instead, I don’t get it.
Thank you for your help again!