torch.sgn

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

Эта функция является расширением torch.sign() для комплексных тензоров. Она вычисляет новый тензор, элементы которого имеют те же углы, что и соответствующие элементы input , и абсолютные значения (т.е. величины) равные единице для комплексных тензоров и эквивалентна torch.sign() для некомплексных тензоров.

$$\text{out}_{i} = \begin{cases} 0 & |\text{{input}}_i| == 0 \\ \frac{{\text{{input}}_i}}{|{\text{{input}}_i}|} & \text{иначе} \end{cases}$$

Параметры

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

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

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

Пример:

>>> t = torch.tensor([3+4j, 7-24j, 0, 1+2j])
>>> t.sgn()
tensor([0.6000+0.8000j, 0.2800-0.9600j, 0.0000+0.0000j, 0.4472+0.8944j])