Function upsample 

pub fn upsample(
    scale_h: usize,
    scale_w: usize,
) -> impl Fn(SymTensor) -> SymTensor

Nearest-neighbour 2-D upsample.

Output shape: [B, C, H*scale_h, W*scale_w].