I can’t reproduce the issue and get the expected results:
cpu
-0.06411010026931763 0.5944567918777466
0.9358899593353271 0.5944567918777466
cuda:0
-0.06415127217769623 0.5944625735282898
0.9358487129211426 0.5944625735282898