ngraph.opset2.topk

ngraph.opset2.topk(data: Union[_pyngraph.Node, int, float, numpy.ndarray], k: Union[_pyngraph.Node, int, float, numpy.ndarray], axis: int, mode: str, sort: str, name: Optional[str] = None)_pyngraph.Node

Return a node which performs TopK.

Parameters
  • data – Input data.

  • k

  • axis – TopK Axis.

  • mode – Compute TopK largest (‘max’) or smallest (‘min’)

  • sort – Order of output elements (sort by: ‘none’, ‘index’ or ‘value’)

Returns

The new node which performs TopK (both indices and values)