Function median 

pub fn median<T>(input: &Array<T>, dim: i64) -> Array<T>
where
    T: HasAfEnum + MedianComputable,

Find the median along a given dimension

Parameters

• input is the input Array
• dim is dimension along which median has to be found

Return Values

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.