sciai.utils.flatten_add_dim

sciai.utils.flatten_add_dim(*data)[source]

flatten data and add an extra dimension at the end.

Parameters: *data (np.array) – data to flatten and add dimension.
Returns: tuple[np.array], converted data.