Функция torch.is_floating_point в PyTorch
Справочное руководство PyTorch torch
torch.is_floating_pointЭто функция в PyTorch, используемая для проверки, является ли тип данных тензора типом с плавающей запятой.
Определение функции
torch.is_floating_point(input)
Пример использования
Пример
import torch
# Тензор с плавающей запятой
x = torch.tensor([1.0, 2.0, 3.0])
print("Число с плавающей запятой:", torch.is_floating_point(x))
# Целочисленный тензор
y = torch.tensor([1, 2, 3])
print("Целое число:", torch.is_floating_point(y))
# Тензор с плавающей запятой
x = torch.tensor([1.0, 2.0, 3.0])
print("Число с плавающей запятой:", torch.is_floating_point(x))
# Целочисленный тензор
y = torch.tensor([1, 2, 3])
print("Целое число:", torch.is_floating_point(y))
Результат вывода:
浮点数: True 整数: False
Другие расширения