Función torch.is_floating_point de PyTorch
Manual de referencia de Pytorch torch
torch.is_floating_pointEs una función en PyTorch utilizada para comprobar si el tipo de datos de un tensor es de punto flotante.
Definición de la función
torch.is_floating_point(input)
Ejemplo de uso
Ejemplo
import torch
# Tensor de punto flotante
x = torch.tensor([1.0, 2.0, 3.0])
print("Flotante:", torch.is_floating_point(x))
# Tensor entero
y = torch.tensor([1, 2, 3])
print("Entero:", torch.is_floating_point(y))
# Tensor de punto flotante
x = torch.tensor([1.0, 2.0, 3.0])
print("Flotante:", torch.is_floating_point(x))
# Tensor entero
y = torch.tensor([1, 2, 3])
print("Entero:", torch.is_floating_point(y))
El resultado de salida es:
浮点数: True 整数: False
Otras extensiones