Hi @jahanifar,
This is most likely due to the print options settings in pytorch (docs here)
Can you set torch.set_printoptions(precision=10) (just after you import torch) and re-run this? You’ll get the correct values. If not, use torch.float64 instead!