ArgMax

ArgMax - 13

版本

  • 名稱: ArgMax (GitHub)

  • 領域: main

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

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

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

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

    將輸入和輸出類型約束為所有數值張量。