mars.tensor.array_split#

mars.tensor.array_split(a, indices_or_sections, axis=0)[来源]#

将张量拆分为多个子张量。

请参阅 split 文档。这两个函数之间唯一的区别是 array_split 允许 indices_or_sections 为一个不平均划分轴的整数。对于一个长度为 l 的张量，应该被划分为 n 个部分，它返回 l % n 个大小为 l//n + 1 的子数组，剩下的是大小为 l//n。

另请参阅

示例

>>> import mars.tensor as mt

>>> x = mt.arange(8.0)
>>> mt.array_split(x, 3).execute()
    [array([ 0.,  1.,  2.]), array([ 3.,  4.,  5.]), array([ 6.,  7.])]

>>> x = mt.arange(7.0)
>>> mt.array_split(x, 3).execute()
    [array([ 0.,  1.,  2.]), array([ 3.,  4.]), array([ 5.,  6.])]