Función PyTorch torch.get_default_dtype

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

torch.get_default_dtypeEs una función en PyTorch que se utiliza para obtener el tipo de dato de punto flotante predeterminado actual.

Definición de la función

torch.get_default_dtype()

Ejemplo de uso

Ejemplo

import torch

# Obtener el dtype predeterminado
dtype = torch.get_default_dtype()
print("dtype predeterminado:", dtype)

El resultado de salida es:

默认 dtype: torch.float32

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

Otras extensiones