indeed even torch.equal returns true. I don’t know how I got two different(but close) results last time, sorry.