pub fn median(images: Tensor<4>, size: usize) -> Tensor<4>
Apply a depth-wise square median filter to the given images