GP

Results 1 comments of

GP

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

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