下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

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

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词