print_along_axis