dev_unify_array

ivy.dev_unify_array(xs, dev_str, mode, axis=0)[source]

Unify a list of sub-arrays, on arbitrary devices, to a single array on the specified device.

Parameters
  • xs (sequence of arrays) – The list of arrays to unify onto the specified device.

  • dev_str (str) – The device to unify the arrays to.

  • mode (str) – The mode by which to unify, must be one of [ concat | mean | sum ]

  • axis (int, optional) – The axis along which to concattenate the array, if concat mode is set. Default is 0.

Returns

array unified to the target device


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty