torch.isreal

torch.isreal(input) → Tensor

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

Параметры:

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

Возвращаемое значение:

Булевый тензор, равный True, где input вещественный, и False в противном случае

Пример:

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