Trilu

Trilu - 14

版本

  • 名稱: Trilu (GitHub)

  • 網域: main

  • since_version: 14

  • 函式: False

  • 支援等級: SupportType.COMMON

  • 形狀推斷: True

此版本的運算子自版本 14 起可用。

摘要

給定一個 2 維矩陣或多批 2 維矩陣，返回張量的上三角或下三角部分。屬性 "upper" 決定保留上部分還是下部分。如果設定為 true，則保留上三角矩陣。否則保留下三角矩陣。"upper" 屬性的預設值為 true。Trilu 採用一個形狀為 [*, N, M] 的輸入張量，其中 * 是零個或多個批次維度。上三角部分由給定對角線 (k) 上及其上方的元素組成。下三角部分由對角線上的元素及其下方的元素組成。矩陣中的所有其他元素都設定為零。如果 k = 0，則保留主對角線及其上方/下方的三角形部分。如果 upper 設定為 true，則正數 k 會保留上三角矩陣，但不包含主對角線和它上方的 (k-1) 條對角線。負數 k 值會保留主對角線及其下方的 |k| 條對角線。如果 upper 設定為 false，則正數 k 會保留下三角矩陣，包括主對角線和其上方的 k 條對角線。負數 k 值不包括主對角線及其下方的 (|k|-1) 條對角線。

屬性

  • upper - INT (預設值為 '1')

    布林值。表示保留矩陣的上部分還是下部分。預設值為 true。

輸入

介於 1 和 2 個輸入之間。

  • input (異質) - T

    階數為 2 或更高的輸入張量。

  • k (可選，異質) - tensor(int64)

    一個 0 維張量，包含一個單一值，對應於要排除或包含的主對角線上方或下方的對角線數。如果未指定，則預設值為 0。

輸出

  • output (異質) - T

    與輸入張量類型和形狀相同的輸出張量。

型別限制

  • T in ( 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) )

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