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])