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 在 (
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 在 (
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 在 (
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 在 (
tensor(double),tensor(float),tensor(float16),tensor(int16),tensor(int32),tensor(int64),tensor(int8),tensor(uint16),tensor(uint32),tensor(uint64),tensor(uint8)) 中将输入和输出类型限制为所有数值张量。