ngraph.non_zero¶
-
ngraph.
non_zero
(data: Union[_pyngraph.Node, int, float, numpy.ndarray], output_type: str = 'i64', name: Optional[str] = None) → _pyngraph.Node¶ Return the indices of the elements that are non-zero.
- Parameters
data – Input data.
output_type – Output tensor type.
- Returns
The new node which performs NonZero