2022-02-19 14:42:47 +08:00
|
|
|
|
mindspore.nn.Triu
|
|
|
|
|
|
==================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:class:: mindspore.nn.Triu
|
|
|
|
|
|
|
|
|
|
|
|
返回一个Tensor，指定主对角线以下的元素被置为0。
|
|
|
|
|
|
|
|
|
|
|
|
将矩阵元素沿主对角线分为上三角和下三角（包含对角线）。
|
|
|
|
|
|
|
|
|
|
|
|
参数 `k` 控制对角线的选择。若 `k` 为0，则沿主对角线分割并保留上三角所有元素。若 `k` 为正值，则沿主对角线向上选择对角线 `k` ，并保留上三角所有元素。若 `k` 为负值，则沿主对角线向下选择对角线 `k` ，并保留上三角所有元素。
|
|
|
|
|
|
|
2022-07-25 16:47:23 +08:00
|
|
|
|
输入：
|
|
|
|
|
|
- **x** (Tensor) - Triu的输入，任意维度的Tensor，其数据类型为数值型。
|
|
|
|
|
|
- **k** (int) - 对角线的索引。默认值：0。
|
2022-02-19 14:42:47 +08:00
|
|
|
|
|
2022-07-25 16:47:23 +08:00
|
|
|
|
输出：
|
|
|
|
|
|
Tensor，数据类型和shape与 `x` 相同。
|
2022-02-19 14:42:47 +08:00
|
|
|
|
|
2022-07-25 16:47:23 +08:00
|
|
|
|
异常：
|
|
|
|
|
|
- **TypeError** - `k` 不是int。
|
|
|
|
|
|
- **ValueError** - `x` 的shape长度小于1。
|
