2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.func_dstack.rst 743 Bytes
一键复制 编辑 原始数据 按行查看 历史
yide12 提交于 2023-03-23 11:45 . code_docs_master

mindspore.ops.dstack

.. py:function:: mindspore.ops.dstack(inputs)

    将多个Tensor沿着第三维度进行堆叠。

    1-D Tensor :math:`(N,)` 重新排列为 :math:`(1,N,1)` ,2-D Tensor :math:`(M,N)` 重新排列为 :math:`(M,N,1)` 。

    参数:
        - **inputs** (Union(List[Tensor], Tuple[Tensor])) - 一个Tensor序列。除了第三个轴外,所有的
          Tensor必须有相同的shape。如果是1-D或2-D的Tensor,则它们的shape必须相同。

    返回:
        堆叠后的Tensor,其维度至少为3。输出shape与 `numpy.dstack()` 类似。

    异常:
        - **TypeError** - 如果 `inputs` 不是list或tuple。
        - **ValueError** - 如果 `inputs` 为空。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助