Split a tensor into a fixed number of equal pieces along an axis.

The size of a along axis must be divisible by num_splits. Each output piece is wired with a SplitSliceBackward node that scatters its gradient back into the correct slice of the input gradient via insert_axis_slice.

See Also

split_at_op, chunk_op, unbind_op.