I’ve tested this on two ARM devices, with pytorch version 1.4.0 and 1.7.0, respectively, and the issue was present in both cases.
Do you know starting with which pytorch version was this issue fixed?
Thanks.
UPDATE: I’ve just tried pytorch 1.8.0 on the NVIDIA Jetson nano, got the same wrong result.