ArgMin¶
ArgMin - 13¶
版本¶
名称: ArgMin (GitHub)
域:
main
始于版本:
13
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本运算符已可用,始于版本 13。
摘要¶
沿给定轴计算输入张量中最小元素的索引。如果 keepdims 等于 1,则结果张量与输入具有相同的秩。如果 keepdims 等于 0,则结果张量会修剪减少的维度。如果 select_last_index 为 True(默认 False),则当最小元素在输入中出现多次时,选择最后一次出现的索引。否则,选择第一次出现的索引。输出张量类型为整数。
属性¶
轴 - INT(默认为
'0'
)计算 arg 索引的轴。接受的范围是 [-r, r-1],其中 r = rank(data)。
keepdims - INT(默认为
'1'
)是否保留减少的维度,默认 1 表示保留减少的维度。
select_last_index - INT(默认为
'0'
)如果 {name} 在多个索引中出现,是否选择最后一次出现的索引或第一次出现的索引,默认为 False(第一次出现的索引)。
输入¶
数据(异构)- T
输入张量。
输出¶
reduced(异构)- 张量(int64)
具有整数数据类型的减少后的输出张量。
类型约束¶
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)
) 中将输入和输出类型约束为所有数值张量。
ArgMin - 12¶
版本¶
名称: ArgMin (GitHub)
域:
main
始于版本:
12
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本运算符已可用,始于版本 12。
摘要¶
沿给定轴计算输入张量中最小元素的索引。如果 keepdims 等于 1,则结果张量与输入具有相同的秩。如果 keepdims 等于 0,则结果张量会修剪减少的维度。如果 select_last_index 为 True(默认 False),则当最小元素在输入中出现多次时,选择最后一次出现的索引。否则,选择第一次出现的索引。输出张量类型为整数。
属性¶
轴 - INT(默认为
'0'
)计算 arg 索引的轴。接受的范围是 [-r, r-1],其中 r = rank(data)。
keepdims - INT(默认为
'1'
)是否保留减少的维度,默认 1 表示保留减少的维度。
select_last_index - INT(默认为
'0'
)如果 {name} 在多个索引中出现,是否选择最后一次出现的索引或第一次出现的索引,默认为 False(第一次出现的索引)。
输入¶
数据(异构)- T
输入张量。
输出¶
reduced(异构)- 张量(int64)
具有整数数据类型的减少后的输出张量。
类型约束¶
T 在 (
tensor(double)
,tensor(float)
,tensor(float16)
,tensor(int16)
,tensor(int32)
,tensor(int64)
,tensor(int8)
,tensor(uint16)
,tensor(uint32)
,tensor(uint64)
,tensor(uint8)
) 中将输入和输出类型约束为所有数值张量。
ArgMin - 11¶
版本¶
名称: ArgMin (GitHub)
域:
main
始于版本:
11
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本运算符已可用,始于版本 11。
摘要¶
沿给定轴计算输入张量中最小元素的索引。如果 keepdims 等于 1,则结果张量与输入具有相同的秩。如果 keepdims 等于 0,则结果张量会修剪减少的维度。输入张量不能为空。输出张量类型为整数。
属性¶
轴 - INT(默认为
'0'
)计算 arg 索引的轴。接受的范围是 [-r, r-1],其中 r = rank(data)。
keepdims - INT(默认为
'1'
)是否保留减少的维度,默认 1 表示保留减少的维度。
输入¶
数据(异构)- T
输入张量。
输出¶
reduced(异构)- 张量(int64)
具有整数数据类型的减少后的输出张量。
类型约束¶
T 在 (
tensor(double)
,tensor(float)
,tensor(float16)
,tensor(int16)
,tensor(int32)
,tensor(int64)
,tensor(int8)
,tensor(uint16)
,tensor(uint32)
,tensor(uint64)
,tensor(uint8)
) 中将输入和输出类型约束为所有数值张量。
ArgMin - 1¶
版本¶
名称: ArgMin (GitHub)
域:
main
始于版本:
1
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本运算符已可用,始于版本 1。
摘要¶
沿给定轴计算输入张量中最小元素的索引。如果 keepdims 等于 1,则结果张量与输入具有相同的秩。如果 keepdims 等于 0,则结果张量会修剪减少的维度。输出张量类型为整数。
属性¶
轴 - INT(默认为
'0'
)计算 arg 索引的轴。
keepdims - INT(默认为
'1'
)是否保留减少的维度,默认 1 表示保留减少的维度。
输入¶
数据(异构)- T
输入张量。
输出¶
reduced(异构)- 张量(int64)
具有整数数据类型的减少后的输出张量。
类型约束¶
T 在 (
tensor(double)
,tensor(float)
,tensor(float16)
,tensor(int16)
,tensor(int32)
,tensor(int64)
,tensor(int8)
,tensor(uint16)
,tensor(uint32)
,tensor(uint64)
,tensor(uint8)
) 中将输入和输出类型约束为所有数值张量。