torch.isreal

torch.isreal(input) → Tensor

Возвращает новый тензор с логическими элементами, указывающими, является ли каждый элемент input вещественным. Все типы с вещественными значениями считаются вещественными. Комплексные значения считаются вещественными, если их мнимая часть равна 0.

Параметры:

input (Tensor) – входной тензор.

Возвращает:

Логический тензор, содержащий True там, где input является вещественным, и False в остальных случаях

Пример:

>>> torch.isreal(torch.tensor([1, 1+1j, 2+0j]))
tensor([True, False, True])