bug: 		swap_stack_axes was not managing the array indexing properly
comment: 	added doc for swap_stack_axes