Returns an Axis object representing the batch axis

get_default_batch_axis()