dpctl.tensor.sign

dpctl.tensor.sign

sign(x, /, *, out=None, order=’K’)

Computes an indication of the sign of each element x_i of input array x using the signum function.

The signum function returns -1 if x_i is less than 0, 0 if x_i is equal to 0, and 1 if x_i is greater than 0.

Parameters:
  • x (usm_ndarray) – Input array, expected to have a numeric data type.

  • out (Union[usm_ndarray, None], optional) – Output array to populate. Array must have the correct shape and the expected data type.

  • order ("C","F","A","K", optional) – Memory layout of the new output array, if parameter out is None. Default: “K”.

Returns:

An array containing the element-wise result of the signum function. The data type of the returned array is determined by the Type Promotion Rules.

Return type:

usm_ndarray