openvino.runtime.opset12.topk#
- openvino.runtime.opset12.topk(data: Node | int | float | ndarray, k: Node | int | float | ndarray, axis: int, mode: str, sort: str, index_element_type: str = 'i32', stable: bool = False, name: str | None = None) 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’)
index_element_type – Type of output tensor with indices.
stable – Specifies whether the equivalent elements should maintain their relative order from the input tensor during sorting.
- Returns:
The new node which performs TopK