split

ivy.split(x, num_sections=None, axis=0, f=None)[source]

Splits an array into multiple sub-arrays.

Parameters
  • 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.

Returns

A list of sub-arrays.


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty