pub fn median<T>(input: &Array<T>, dim: i64) -> Array<T> where T: HasAfEnum + MedianComputable,
Find the median along a given dimension
An Array whose size is equal to input except along the dimension which
median needs to be found. Array size along dim will be reduced to one.