TopK¶
TopK - 24¶
版本¶
名称: TopK (GitHub)
域:
main
起始版本:
24
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的操作符已可用于版本 24 及以上。
摘要¶
沿指定轴检索前 K 个最大或最小的元素。给定一个形状为 [a_0, a_1, …, a_{n-1}] 的输入张量和一个整数参数 k,返回两个输出
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的值张量,包含沿指定轴的前 k 个元素的值
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的索引张量,包含前 k 个元素(来自原始输入张量的索引)。
如果“largest”为 1(默认值),则返回 k 个最大的元素。
如果“sorted”为 1(默认值),则返回的 k 个元素将进行排序。
如果“sorted”为 0,则返回的“Values”和“Indices”的顺序不确定。
给定两个等价的值,此运算符使用沿轴的索引作为平局解决器。也就是说,索引较低的元素将首先出现。
属性¶
轴 - INT(默认值为
'-1'
)进行排序的维度。负值表示从后往前计算维度。可接受的范围是 [-r, r-1],其中 r = rank(input)。
largest - INT (默认值为
'1'
)是否返回前 K 个最大或最小的元素。
sorted - INT (默认值为
'1'
)是否以排序顺序返回元素。
输入¶
X (异构) - T
形状为 [a_0, a_1, …, a_{n-1}] 的张量
K (异构) - tensor(int64)
一个包含单个正值的 1-D 张量,表示要检索的前 k 个元素的数量
输出¶
Values (异构) - T
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的张量,包含输入张量的前 K 个值
Indices (异构) - I
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的张量,包含前 K 个值对应的原始输入张量索引。
类型约束¶
T 在 (
tensor(bfloat16)
,tensor(double)
,tensor(float)
,tensor(float16)
,tensor(int16)
,tensor(int32)
,tensor(int64)
,tensor(int8)
,tensor(uint16)
,tensor(uint32)
,tensor(uint64)
,tensor(uint8)
)将输入和输出类型限制为数字张量。
I 在 (
tensor(int64)
)将索引张量限制为 int64
TopK - 11¶
版本¶
名称: TopK (GitHub)
域:
main
起始版本:
11
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的运算符自 版本 11 起可用。
摘要¶
沿指定轴检索前 K 个最大或最小的元素。给定一个形状为 [a_0, a_1, …, a_{n-1}] 的输入张量和一个整数参数 k,返回两个输出
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的值张量,包含沿指定轴的前 k 个元素的值
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的索引张量,包含前 k 个元素(来自原始输入张量的索引)。
如果“largest”为 1(默认值),则返回 k 个最大的元素。
如果“sorted”为 1(默认值),则返回的 k 个元素将进行排序。
如果“sorted”为 0,则返回的“Values”和“Indices”的顺序不确定。
给定两个等价的值,此运算符使用沿轴的索引作为平局解决器。也就是说,索引较低的元素将首先出现。
属性¶
轴 - INT(默认值为
'-1'
)进行排序的维度。负值表示从后往前计算维度。可接受的范围是 [-r, r-1],其中 r = rank(input)。
largest - INT (默认值为
'1'
)是否返回前 K 个最大或最小的元素。
sorted - INT (默认值为
'1'
)是否以排序顺序返回元素。
输入¶
X (异构) - T
形状为 [a_0, a_1, …, a_{n-1}] 的张量
K (异构) - tensor(int64)
一个包含单个正值的 1-D 张量,表示要检索的前 k 个元素的数量
输出¶
Values (异构) - T
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的张量,包含输入张量的前 K 个值
Indices (异构) - I
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的张量,包含前 K 个值对应的原始输入张量索引。
类型约束¶
T 在 (
tensor(double)
,tensor(float)
,tensor(float16)
,tensor(int16)
,tensor(int32)
,tensor(int64)
,tensor(int8)
,tensor(uint16)
,tensor(uint32)
,tensor(uint64)
,tensor(uint8)
)将输入和输出类型限制为数字张量。
I 在 (
tensor(int64)
)将索引张量限制为 int64
TopK - 10¶
版本¶
名称: TopK (GitHub)
域:
main
since_version:
10
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的操作符已在 版本 10 中提供。
摘要¶
沿指定轴检索前 K 个元素。给定一个形状为 [a_0, a_1, …, a_{n-1}] 的输入张量和一个整数参数 k,返回两个输出:-形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的值张量,包含沿指定轴的前 k 个元素的值 -形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的索引张量,包含前 k 个元素(来自原始输入张量的索引)。
给定两个等价的值,此运算符使用沿轴的索引作为平局解决器。也就是说,索引较低的元素将首先出现。
属性¶
轴 - INT(默认值为
'-1'
)进行排序的维度。
输入¶
X (异构) - T
形状为 [a_0, a_1, …, a_{n-1}] 的张量
K (异构) - tensor(int64)
一个包含单个正值的 1-D 张量,表示要检索的前 k 个元素的数量
输出¶
Values (异构) - T
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的张量,包含输入张量的前 K 个值
Indices (异构) - I
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的张量,包含前 K 个值对应的原始输入张量索引。
类型约束¶
T 在 (
tensor(double)
,tensor(float)
,tensor(float16)
)将输入和输出类型限制为浮点张量。
I 在 (
tensor(int64)
)将索引张量限制为 int64
TopK - 1¶
版本¶
名称: TopK (GitHub)
域:
main
起始版本:
1
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的运算符自 版本 1 起可用。
摘要¶
沿指定轴检索前 K 个元素。给定一个形状为 [a_0, a_1, …, a_{n-1}] 的输入张量和一个整数参数 k,返回两个输出:-形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的值张量,包含沿指定轴的前 k 个元素的值 -形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的索引张量,包含前 k 个元素(来自原始输入张量的索引)。给定两个等价的值,此运算符使用沿轴的索引作为平局解决器。也就是说,索引较低的元素将首先出现。
属性¶
轴 - INT(默认值为
'-1'
)进行排序的维度。
k - INT (必需)
要检索的前 k 个元素的数量
输入¶
X (异构) - T
形状为 [a_0, a_1, …, a_{n-1}] 的张量
输出¶
Values (异构) - T
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的张量,包含输入张量的前 K 个值
Indices (异构) - I
形状为 [a_0, a_1, …, a_{axis-1}, k, a_{axis+1}, … a_{n-1}] 的张量,包含前 K 个值对应的原始输入张量索引。
类型约束¶
T 在 (
tensor(double)
,tensor(float)
,tensor(float16)
)将输入和输出类型限制为浮点张量。
I 在 (
tensor(int64)
)将索引张量限制为 int64