Function normalize_axis 

pub fn normalize_axis(axis: i64, ndim: usize) -> i64

Normalize a potentially negative axis to a positive axis.

For example, normalize_axis(-1, 4) returns 3.