torch (version 0.14.2)

torch_sgn: Sgn

Description

Sgn

Usage

torch_sgn(self)

Arguments

self

(Tensor) the input tensor.

sgn(input, *, out=None) -> Tensor

For complex tensors, this function returns a new tensor whose elemants have the same angle as that of the elements of input and absolute value 1. For a non-complex tensor, this function returns the signs of the elements of input (see torch_sign).

${\text{out}}_i=0$, if $|{\text{{input}}}_i|==0$ ${\text{out}}_i=\frac{{\text{{input}}}_i}{|{\text{{input}}}_i|}$, otherwise

Examples

if (torch_is_installed()) {
if (FALSE) {
x <- torch_tensor(c(3+4i, 7-24i, 0, 1+2i))
x$sgn()
torch_sgn(x)
}
}