Table Of Contents
Table Of Contents

mxnet.gluon.utils.split_data

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

Splits an NDArray into num_slice slices along batch_axis. Usually used for data parallelism where each slices is sent to one device (i.e. GPU).

Parameters

  • data (NDArray) – A batch of data.

  • num_slice (int) – Number of desired slices.

  • 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. If True, an error will be raised when num_slice does not evenly divide data.shape[batch_axis].

Returns

Return value is a list even if num_slice is 1.

Return type

list of NDArray

Previous
utils
Next
mxnet.gluon.utils.split_and_load