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