Как использовать double как тип по умолчанию для плавающих чисел в PyTorch

Question

Я хочу, чтобы все плавающие числа в моем коде PyTorch типа double по умолчанию, как я могу это сделать?

Answer 1

Вы должны использовать для этого torch.set_default_dtype.

Это правда, что использование torch.set_default_tensor_type также будет иметь аналогичный эффект, но torch.set_default_tensor_type не только устанавливает тип данных по умолчанию, но также устанавливает значения по умолчанию для устройства , где расположен тензор и макет тензора.

Answer 2

Вы ищете torch.set_default_tensor_type:

torch.set_default_tensor_type(torch.DoubleTensor)

Вы можете использовать torch.set_default_dtype:

torch.set_default_dtype(torch.float64)