Flatten (攤平)

Flatten - 23

版本

  • 名稱: Flatten (GitHub)

  • 網域: main

  • since_version: 23

  • function: False

  • support_level: SupportType.COMMON

  • 形狀推斷: True

此版本的運算子自第 23 版開始提供。

摘要

將輸入張量攤平成一個 2D 矩陣。如果輸入張量的形狀為 (d_0, d_1, … d_n)，則輸出將具有形狀 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

屬性

  • 軸 - INT (預設為 '1')

    指示應將哪個輸入維度（不包含）攤平到輸出的外維度。軸的值必須在 [-r, r] 範圍內，其中 r 是輸入張量的秩。負值表示從後往前計算維度。當軸 = 0 時，輸出張量的形狀為 (1, (d_0 X d_1 … d_n)，其中輸入張量的形狀為 (d_0, d_1, … d_n)。

輸入

  • 輸入 (異質) - T

    秩 >= 軸的張量。

輸出

  • 輸出 (異質) - T

    一個 2D 張量，其內容與輸入張量相同，其中輸入維度直到軸都被攤平到輸出的外維度，而剩餘的輸入維度則被攤平到輸出的內維度。

類型限制

  • T 在 ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(float4e2m1), tensor(float8e4m3fn), tensor(float8e4m3fnuz), tensor(float8e5m2), tensor(float8e5m2fnuz), tensor(int16), tensor(int32), tensor(int4), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint4), tensor(uint64), tensor(uint8) )

    將輸入和輸出限制為所有張量類型，直到 IRv10。

Flatten - 21

版本

  • 名稱: Flatten (GitHub)

  • 網域: main

  • since_version: 21

  • function: False

  • support_level: SupportType.COMMON

  • 形狀推斷: True

此版本的運算子自第 21 版開始提供。

摘要

將輸入張量攤平成一個 2D 矩陣。如果輸入張量的形狀為 (d_0, d_1, … d_n)，則輸出將具有形狀 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

屬性

  • 軸 - INT (預設為 '1')

    指示應將哪個輸入維度（不包含）攤平到輸出的外維度。軸的值必須在 [-r, r] 範圍內，其中 r 是輸入張量的秩。負值表示從後往前計算維度。當軸 = 0 時，輸出張量的形狀為 (1, (d_0 X d_1 … d_n)，其中輸入張量的形狀為 (d_0, d_1, … d_n)。

輸入

  • 輸入 (異質) - T

    秩 >= 軸的張量。

輸出

  • 輸出 (異質) - T

    一個 2D 張量，其內容與輸入張量相同，其中輸入維度直到軸都被攤平到輸出的外維度，而剩餘的輸入維度則被攤平到輸出的內維度。

類型限制

  • T 在 ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(float8e4m3fn), tensor(float8e4m3fnuz), tensor(float8e5m2), tensor(float8e5m2fnuz), tensor(int16), tensor(int32), tensor(int4), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint4), tensor(uint64), tensor(uint8) )

    將輸入和輸出限制為所有張量類型，直到 IRv10。

Flatten - 13

版本

  • 名稱: Flatten (GitHub)

  • 網域: main

  • since_version: 13

  • function: False

  • support_level: SupportType.COMMON

  • 形狀推斷: True

這個運算子的版本從版本 13 開始提供。

摘要

將輸入張量攤平成一個 2D 矩陣。如果輸入張量的形狀為 (d_0, d_1, … d_n)，則輸出將具有形狀 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

屬性

  • 軸 - INT (預設為 '1')

    指示應將哪個輸入維度（不包含）攤平到輸出的外維度。軸的值必須在 [-r, r] 範圍內，其中 r 是輸入張量的秩。負值表示從後往前計算維度。當軸 = 0 時，輸出張量的形狀為 (1, (d_0 X d_1 … d_n)，其中輸入張量的形狀為 (d_0, d_1, … d_n)。

輸入

  • 輸入 (異質) - T

    秩 >= 軸的張量。

輸出

  • 輸出 (異質) - T

    一個 2D 張量，其內容與輸入張量相同，其中輸入維度直到軸都被攤平到輸出的外維度，而剩餘的輸入維度則被攤平到輸出的內維度。

類型約束

  • T 在 ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

    限制輸入和輸出為所有張量類型。

Flatten - 11

版本

  • 名稱: Flatten (GitHub)

  • 網域: main

  • since_version: 11

  • function: False

  • support_level: SupportType.COMMON

  • 形狀推斷: True

這個運算子的版本從版本 11 開始提供。

摘要

將輸入張量攤平成一個 2D 矩陣。如果輸入張量的形狀為 (d_0, d_1, … d_n)，則輸出將具有形狀 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

屬性

  • 軸 - INT (預設為 '1')

    指示應將哪個輸入維度（不包含）攤平到輸出的外維度。軸的值必須在 [-r, r] 範圍內，其中 r 是輸入張量的秩。負值表示從後往前計算維度。當軸 = 0 時，輸出張量的形狀為 (1, (d_0 X d_1 … d_n)，其中輸入張量的形狀為 (d_0, d_1, … d_n)。

輸入

  • 輸入 (異質) - T

    秩 >= 軸的張量。

輸出

  • 輸出 (異質) - T

    一個 2D 張量，其內容與輸入張量相同，其中輸入維度直到軸都被攤平到輸出的外維度，而剩餘的輸入維度則被攤平到輸出的內維度。

類型約束

  • T 在 ( tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

    限制輸入和輸出為所有張量類型。

Flatten - 9

版本

  • 名稱: Flatten (GitHub)

  • 網域: main

  • since_version: 9

  • function: False

  • support_level: SupportType.COMMON

  • 形狀推斷: True

這個運算子的版本從版本 9 開始提供。

摘要

將輸入張量攤平成一個 2D 矩陣。如果輸入張量的形狀為 (d_0, d_1, … d_n)，則輸出將具有形狀 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

屬性

  • 軸 - INT (預設為 '1')

    指出應該將輸入維度（不包含）展平到輸出外維度的上限。軸的值必須在 [0, R] 的範圍內，其中 R 是輸入張量的秩。當 axis = 0 時，輸出張量的形狀為 (1, (d_0 X d_1 … d_n)，其中輸入張量的形狀為 (d_0, d_1, … d_n)。

輸入

  • 輸入 (異質) - T

    秩 >= 軸的張量。

輸出

  • 輸出 (異質) - T

    一個 2D 張量，其內容與輸入張量相同，其中輸入維度直到軸都被攤平到輸出的外維度，而剩餘的輸入維度則被攤平到輸出的內維度。

類型約束

  • T 在 ( tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

    限制輸入和輸出為所有張量類型。

Flatten - 1

版本

  • 名稱: Flatten (GitHub)

  • 網域: main

  • since_version: 1

  • function: False

  • support_level: SupportType.COMMON

  • 形狀推斷: True

這個運算子的版本從版本 1 開始提供。

摘要

將輸入張量攤平成一個 2D 矩陣。如果輸入張量的形狀為 (d_0, d_1, … d_n)，則輸出將具有形狀 (d_0 X d_1 … d_(axis-1), d_axis X d_(axis+1) … X dn)。

屬性

  • 軸 - INT (預設為 '1')

    指出應該將輸入維度（不包含）展平到輸出外維度的上限。軸的值必須在 [0, R] 的範圍內，其中 R 是輸入張量的秩。當 axis = 0 時，輸出張量的形狀為 (1, (d_0 X d_1 … d_n)，其中輸入張量的形狀為 (d_0, d_1, … d_n)。

輸入

  • 輸入 (異質) - T

    秩 >= 軸的張量。

輸出

  • 輸出 (異質) - T

    一個 2D 張量，其內容與輸入張量相同，其中輸入維度直到軸都被攤平到輸出的外維度，而剩餘的輸入維度則被攤平到輸出的內維度。

類型約束

  • T 在 ( tensor(double), tensor(float), tensor(float16) )

    將輸入和輸出類型限制為浮點張量。