squeeze_axis=1 removes the axis with length 1 from the shapes of the output arrays.
Note that setting squeeze_axis to 1 removes axis with length 1 only
along the axis which it is split.
Also squeeze_axis can be set to true only if input.shape[axis]==num_outputs.

num_outputs (int, required) – Number of splits. Note that this should evenly divide the length of the axis.

axis (int, optional, default='1') – Axis along which to split.

squeeze_axis (boolean, optional, default=0) – If true, Removes the axis with length 1 from the shapes of the output arrays. Note that setting squeeze_axis to true removes axis with length 1 only along the axis which it is split. Also squeeze_axis can be set to true only if input.shape[axis]==num_outputs.