Función torch.set_default_dtype de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.set_default_dtypeEs una función en PyTorch utilizada para establecer el tipo de dato de punto flotante predeterminado.

Definición de la función

torch.set_default_dtype(d)

Ejemplo de uso

Ejemplo

import torch

# Establecer el dtype predeterminado a float64
torch.set_default_dtype(torch.float64)

# Crear un tensor, usando float64 por defecto
x = torch.tensor([1.0, 2.0, 3.0])
print("dtype predeterminado:", x.dtype)

# Restaurar el predeterminado
torch.set_default_dtype(torch.float32)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones