torch (version 0.0.2)

torch_sign: Sign

Description

Sign

Arguments

input

(Tensor) the input tensor.

out

(Tensor, optional) the output tensor.

sign(input, out=None) -> Tensor

Returns a new tensor with the signs of the elements of input.

$$ \mbox{out}_{i} = \mbox{sgn}(\mbox{input}_{i}) $$

Examples

# NOT RUN {
if (torch_is_installed()) {

a = torch_tensor(c(0.7, -1.2, 0., 2.3))
a
torch_sign(a)
}
# }