Table Of Contents
Table Of Contents


mxnet.gluon.utils.split_and_load(data, ctx_list, batch_axis=0, even_split=True)[source]

Splits an NDArray into len(ctx_list) slices along batch_axis and loads each slice to one context in ctx_list.

  • data (NDArray) – A batch of data.

  • ctx_list (list of Context) – A list of Contexts.

  • batch_axis (int, default 0) – The axis along which to slice.

  • even_split (bool, default True) – Whether to force all slices to have the same number of elements.


Each corresponds to a context in ctx_list.

Return type

list of NDArray