Splits a Variable into n chunks along the specified axis.

In case a tensor cannot be split evenly, with la == length_axis, n = n_chunks it returns la mod n subtensors of size (la div n) + 1 the rest of size la div n. So split sizes at most differs by 1

This is consistent with numpy array_split

Join a sequence of Variables along a new axis into a new Variable. All variables must be of the same shape

Input:

• a variable
• an axis (dimension)

Returns:

• a new stacked variable along the new axis