sampling_locations = (ref + grid * offset_scale).repeat(N_, 1, 1, 1, 1).flatten(3, 4) + \ RuntimeError: The size of tensor a (16) must match the size of tensor b (32) at non-singleton dimension 2

[Result1]

tensor at non-singleton dimension 2

[FindOne1]

#245 (comment)

It's still throwing this error.

[gaodaren47297]

so，brother，have you already handled this problem？Now，I come across the same problem。

[Ducking-WJ]

I'm having a similar issue: sampling_locations = (ref + grid * offset_scale).repeat(N_, 1, 1, 1, 1).flatten(3, 4) +
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: The size of tensor a (86) must match the size of tensor b (256) at non-singleton dimension 2