float32 tensors with torch 1.9.0

This version creates tensors as float32 and works with torch 1.9.0. Due to some numerical instability issue, this does not work with torch 1.10.0 and above.