2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.scipy.linalg.block_diag.rst 807 Bytes
一键复制 编辑 原始数据 按行查看 历史
俞涵 提交于 2024-01-23 16:56 . [i] modify format

mindspore.scipy.linalg.block_diag

.. py:function:: mindspore.scipy.linalg.block_diag(*arrs)

    根据输入的数组创建块对角矩阵。

    输入为:`A`、`B` 和 `C` 的Tensor列表。输出为:在对角线上排列这些Tensor的块对角矩阵。

    .. code-block::

        [[A, 0, 0],
         [0, B, 0],
         [0, 0, C]]

    .. note::
        Windows平台上还不支持 `block_diag`。

    参数:
        - **arrs** (list) - 最大支持2D的Tensor输入。
          一个或多个Tensor,维度支持0D,1D、2D。

    返回:
        对角线上含有 `A`、`B`、`C`,...的Tensor,数据类型与 `A` 相同。

    异常:
        - **ValueError** - 输入参数中存在维度大于2的Tensor。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助