mars.tensor.flatnonzero#
- mars.tensor.flatnonzero(a)[source]#
Return indices that are non-zero in the flattened version of a.
This is equivalent to a.ravel().nonzero()[0].
- Parameters
a (Tensor) – Input tensor.
- Returns
res – Output tensor, containing the indices of the elements of a.ravel() that are non-zero.
- Return type
Tensor
See also
Examples
>>> import mars.tensor as mt
>>> x = mt.arange(-2, 3) >>> x.execute() array([-2, -1, 0, 1, 2]) >>> mt.flatnonzero(x).execute() array([0, 1, 3, 4])
Use the indices of the non-zero elements as an index array to extract these elements:
>>> x.ravel()[mt.flatnonzero(x)].execute() # TODO(jisheng): accomplish this after fancy indexing is supported