torch.signbit

torch.signbit(input, *, out=None) → Tensor

Проверяет, установлен ли знаковый бит каждого элемента input.

Параметры

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

Ключевые аргументы

out (Tensor, необязательно) – выходной тензор.

Пример:

>>> a = torch.tensor([0.7, -1.2, 0., 2.3])
>>> torch.signbit(a)
tensor([ False, True,  False,  False])
>>> a = torch.tensor([-0.0, 0.0])
>>> torch.signbit(a)
tensor([ True,  False])

Примечание

Функция signbit обрабатывает знаковое ноль, поэтому отрицательное ноль (-0) возвращает True.