dev_unify(xs, dev_str, mode, axis=0)¶
Unify a list of sub-arrays, on arbitrary devices, to a single concattenated array on the specified device.
xs (sequence of arrays) – The list of sub-arrays to unify onto the specified device.
dev_str (str) – The device to unify the sub-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.
array unified to the target device