dev_dist_iter

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

Distribute elements of the iterbale xs across the specified devices.

Parameters
  • xs (iterable of any) – The iterable of items to distribute.

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

  • axis (int, optional) – The axis along which to split the arrays in the iterable xs. Default is 0.

Returns

iterable with each element distributed to the target devices


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty