PyTorch torch.sign Function
PyTorch torch Reference Manual
torch.signIt is a function in PyTorch used to return the sign element-wise. It returns -1 (negative), 0 (zero), or 1 (positive).
Function Definition
torch.sign(input, out=None)
Usage Examples
Example
import torch
# Return the sign element-wise
x = torch.tensor([-2.0, 0.0, 3.0])
result = torch.sign(x)
print(result)
# Return the sign element-wise
x = torch.tensor([-2.0, 0.0, 3.0])
result = torch.sign(x)
print(result)
Other Extensions