TopK

功能

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

边界

【输入】

2个输入

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

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