xdas.parallel.parallelize#

xdas.parallel.parallelize(split_axis=0, concat_axis=0, parallel=None)[source]#

Split array positional arguments across threads.

Parameters:

  • split_axis (int or tuple of int, optional) – Axis (or axes) along which to split positional array arguments. Use None for arguments that should not be split.

  • concat_axis (int or tuple of int, optional) – Axis (or axes) along which to concatenate the per-worker outputs.

  • parallel (int, bool, or None, optional) – Worker count override. Forwarded to get_workers_count().

Returns:

decorator – A function decorator.

Return type:

callable