mindspore.numpy.heaviside
- mindspore.numpy.heaviside(x1, x2, dtype=None)[source]
Computes the Heaviside step function.
Note
Numpy arguments out, where, casting, order, subok, signature, and extobj are not supported.
- Parameters
x1 (Tensor) – Input values.
x2 (Tensor) – The value of the function when x1 is 0. If
x1.shape != x2.shape
, they must be broadcastable to a common shape (which becomes the shape of the output).dtype (
mindspore.dtype
, optional) – defaults to None. Overrides the dtype of the output Tensor.
- Returns
Tensor or scalar, the output array, element-wise Heaviside step function of x1. This is a scalar if both x1 and x2 are scalars.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import mindspore.numpy as np >>> output = np.heaviside(np.array([-1.5, 0, 2.0]), np.array(0.5)) >>> print(output) [0. 0.5 1. ] >>> output = np.heaviside(np.array([-1.5, 0, 2.0]), np.array(1)) >>> print(output) [0. 1. 1.]