mindspore/docs/api/api_python/ops/mindspore.ops.func_stack.rst

13 lines
316 B
ReStructuredText

mindspore.ops.stack
====================
.. py:function:: mindspore.ops.stack(tensors, axis=0)
在指定轴上对输入的tensors进行堆叠。
参数:
- **tensors** (Union[tuple, list]) - 输入tensors。
- **axis** (int) - 指定轴,默认 ``0``
返回:
Tensor