megengine.functional.split¶
- split(inp, nsplits_or_sections, axis=0)[源代码]¶
把一个张量分隔成很多个小张量。当`nsplits_or_sections`是整数时，最后一个张量可能比其他张量小。
- 参数
inp – 输入张量。
nsplits_or_sections – 子张量或每个部分的信息列表的个数。
axis – 需要被分隔的轴。
- 返回
输出张量列表。
实际案例
>>> import os >>> import numpy as np >>> x = Tensor(np.random.random((10, 20)), dtype=np.float32) >>> y = F.split(x, 3) >>> z = F.split(x, [6, 17], axis=1) >>> print([i.numpy().shape for i in y]) [(4, 20), (3, 20), (3, 20)] >>> print([i.numpy().shape for i in z]) [(10, 6), (10, 11), (10, 3)]