ArgMax

ArgMax - 13

版本

  • 名称: ArgMax (GitHub)

  • 域名: main

  • since_version: 13

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本13起可用。

摘要

计算输入张量元素沿指定轴的最大元素的索引。如果keepdims等于1，则结果张量的秩与输入相同。如果keepdims等于0，则结果张量的维度会被缩减。如果select_last_index为True（默认为False），则在输入中最大元素出现多次时，选择最后一次出现的索引。否则，选择第一次出现的索引。输出张量的类型为整数。

属性

  • axis - INT (默认为 '0'):

    计算参数索引的轴。接受的范围是[-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

  • since_version: 12

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本12起可用。

摘要

计算输入张量元素沿指定轴的最大元素的索引。如果keepdims等于1，则结果张量与输入张量的秩相同。如果keepdims等于0，则结果张量的维度会减少。如果select_last_index为True（默认为False），则在输入中最大元素出现多次时，选择最后一次出现的索引。否则，选择第一次出现的索引。输出张量的类型为整数。

属性

  • axis - INT (默认为 '0'):

    计算参数索引的轴。接受的范围是[-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

  • since_version: 11

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本11起可用。

总结

计算输入张量元素沿指定轴的最大元素的索引。如果keepdims等于1，则结果张量与输入张量具有相同的秩。如果keepdims等于0，则结果张量的维度会减少。输入张量不能为空。输出张量的类型为整数。

属性

  • axis - INT (默认为 '0'):

    计算参数索引的轴。接受的范围是[-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

  • since_version: 1

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本1起可用。

摘要

计算输入张量元素沿指定轴的最大元素的索引。如果keepdims等于1，则结果张量与输入张量的秩相同。如果keepdims等于0，则结果张量的维度会减少。输出张量的类型为整数。

属性

  • axis - INT (默认为 '0'):

    计算参数索引的轴。

  • 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) ):

    将输入和输出类型限制为所有数值张量。