Как сделать все переменные, созданные в файле PyTorch, в float64? Есть ли одна строка кода, которая может это сделать?
Вы можете установить тип тензора по умолчанию, используя эту строку:
torch.set_default_tensor_type(torch.DoubleTensor)