17 lines
557 B
ReStructuredText
17 lines
557 B
ReStructuredText
mindspore.ops.dstack
|
||
====================
|
||
|
||
.. py:function:: mindspore.ops.dstack(tensors)
|
||
|
||
将多个tensor沿着第三维度进行堆叠。
|
||
|
||
.. note::
|
||
- 一维tensor :math:`(N,)` 重新排列为 :math:`(1,N,1)` ，二维tensor :math:`(M,N)` 重新排列为 :math:`(M,N,1)` 。
|
||
- 除了第三个轴外，所有的tensor必须有相同的shape。如果是一维或二维tensor，则它们的shape必须相同。
|
||
|
||
参数：
|
||
- **tensors** (Union(List[Tensor], tuple[Tensor])) - 输入tensors。
|
||
|
||
返回：
|
||
Tensor
|