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