maxframe.tensor.array_split

maxframe.tensor.array_split(a, indices_or_sections, axis=0)

将一个张量分割成多个子张量。

请参考 split 的文档。这两个函数之间的唯一区别是 array_split 允许 indices_or_sections 是一个不能整除轴的整数。对于长度为 l 的张量，若要将其分割成 n 个部分，则会返回 l % n 个大小为 l//n + 1 的子数组，其余的大小为 l//n。

参见

split

将张量分割成多个大小相等的子张量。

示例

>>> import maxframe.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.])]