ivy.dev_dist_array(x, dev_strs: Union[Iterable[str], Dict[str, int]], axis=0)[source]

Distribute an array across the specified devices, returning a list of sub-arrays, each on a different device.

  • x (array) – The array to distribute across devices.

  • dev_strs (sequence of strs or dict of split sizes) – The devices to distribute the array across.

  • axis (int, optional) – The axis along which to split the array. Default is 0.


array distributed across the target devices

Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty