TopK

功能

返回指定轴的k个最大或最小值。

输入

• ONNX版本为Opset < 10

  一个输入：

  x：tensor，数据类型：float16、float。

• ONNX版本为Opset >= 10

  两个输入：

  x：tensor，数据类型：float16、float。

  k：tensor，数据类型：int64。

输出

两个输出：

Values：topk的返回值。

Indices：topk的返回值索引。

属性

axis：int，默认为-1，含义：指定排序的轴。

largest：int，默认为1，含义：返回k个最大/最小值。

sorted：int，默认为1，含义：是否升序。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18