16 lines
973 B
ReStructuredText
16 lines
973 B
ReStructuredText
mindspore.numpy.dsplit
|
||
=================================
|
||
|
||
.. py:function:: mindspore.numpy.dsplit(x, indices_or_sections)
|
||
|
||
沿第三轴（深度）将Tensor分割为多个sub-tensor。此操作等同于使用 :math:`axis=2` （默认值）进行分割，不论数组的维度如何，始终沿第三轴分割。
|
||
|
||
参数：
|
||
- **x** (Tensor) - 待分割的Tensor。
|
||
- **indices_or_sections** (Union[int, tuple(int), list(int)]) - 如果是整数 :math:`N` ，Tensor将沿轴分割为 :math:`N` 个相等的sub-tensor。如果是tuple(int)、list(int)或排序后的整数，则指示沿轴的分割位置。例如，对于 :math:`axis=0` ， :math:`[2,3]` 将产生三个sub-tensor： :math:`x[:2]` 、 :math:`x[2:3]` 和 :math:`x[3:]` 。如果索引超出轴上数组的维度，则相应地返回空子数组。
|
||
|
||
返回：
|
||
sub-tensor列表。
|
||
|
||
异常：
|
||
- **TypeError** - 如果参数 ``indices_or_sections`` 不是整数。 |