Pytorch Float32 Format