split(x, num_sections=None, axis=0, f=None)¶
Splits an array into multiple sub-arrays.
x (array) – Tensor to be divided into sub-arrays.
num_sections (int, optional) – Number of equal arrays to divide the array into, along the given axis. Size of axis dimension must be multiple of num_sections. Default is to divide into as many arrays as the axis dimension.
axis (int, optional) – The axis along which to split, default is 0.
f (ml_framework, optional) – Machine learning framework. Inferred from inputs if None.
A list of sub-arrays.