2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.func_tril.rst 889 Bytes
一键复制 编辑 原始数据 按行查看 历史
wind-zyx 提交于 2023-04-21 16:49 . ops update value format 0421

mindspore.ops.tril

.. py:function:: mindspore.ops.tril(input, diagonal=0)

    返回输入Tensor `input` 的下三角形部分(包含对角线和下面的元素),并将其他元素设置为0。

    参数:
        - **input** (Tensor) - 输入Tensor。shape为 :math:`(x_1, x_2, ..., x_R)` ,其rank至少为2。
          支持的数据类型有包括所有数值型和bool类型。
        - **diagonal** (int,可选) - 指定对角线位置,默认值: ``0`` ,指定主对角线。

    返回:
        Tensor,其数据类型和shape维度与 `input` 相同。

    异常:
        - **TypeError** - 如果 `input` 不是Tensor。
        - **TypeError** - 如果 `diagonal` 不是int类型。
        - **TypeError** - 如果 `input` 的数据类型既不是数值型也不是bool。
        - **ValueError** - 如果 `input` 的秩小于2。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.2

搜索帮助