Trilu

Trilu - 14

版本

  • 名称: Trilu (GitHub)

  • : main

  • 起始版本: 14

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的操作符自版本 14 起可用。

摘要

给定一个二维矩阵或二维矩阵的批次,返回张量(或张量批次)的上三角部分或下三角部分。“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 个输入之间。

  • 输入 (异构) - T

    秩为 2 或更高的输入张量。

  • k (可选,异构) - tensor(int64)

    一个 0 维张量,包含一个对应于主对角线上方或下方要排除或包含的对角线数量的单个值。如果未指定,则默认值为 0。

输出

  • 输出 (异构) - T

    输出张量,其类型和形状与输入张量相同。

类型约束

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

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