sign

→Tensor

sign(input: Tensor)

Element-wise sign function.

Returns -1 for negative entries, 0 for zero, and +1 for positive entries.

inputTensor

Input tensor.

Tensor

Element-wise result with the same shape as input.

Mathematical definition:

\text{out}_i = \mathrm{sgn}(\text{input}_i) = \begin{cases} -1 & x_i < 0 \\ 0 & x_i = 0 \\ +1 & x_i > 0 \end{cases}

Gradient is zero almost everywhere (treated as zero in autograd).

>>> import lucid
>>> x = lucid.tensor([1.0, 2.0, 3.0])
>>> lucid.sign(x)
Tensor([...])

Loading…