TopK

功能

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

边界

【输入】（ONNX版本为Opset < 10）

1个输入

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

【输入】（ONNX版本为Opset >= 10）

2个输入

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

k：tensor，数据类型：int64

【输出】

2个输出

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