Функция torch.isreal в PyTorch


Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.isrealЭто функция в PyTorch, используемая для проверки, является ли значение вещественным числом. Она возвращает логический тензор, в котором для каждого элемента значение равно True, если оно является вещественным числом (не комплексным числом).

Определение функции

torch.isreal(input)

Пример использования

Пример

import torch

# Создать тензор, содержащий вещественные и комплексные числа
x = torch.tensor([1.0, 2+3j, 4.0, 5+1j])

# Проверить, является ли значение вещественным числом
result = torch.isreal(x)
print(result)

Результат вывода:

tensor([True, False, True, False])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения