return torch.float64 if device is not mps or xpu, else return torch.float32
Attach a file by drag & drop or click to upload