ngraph.opset6.non_zero

ngraph.opset6.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