node_dist

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

Distribute the input item across the specified nodes, returning a list of sub-items, each on a different node.

Parameters
  • x (array or container) – The input array or container to distribute across nodes.

  • node_strs (sequence of strs or dict of split sizes) – The nodes to distribute the input across.

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

Returns

array or container distributed across the target nodes


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty