Shall re rename it reshape_num, reshape_channels, reshape_height and reshape_width to avoid confusion? Since num, channels, height and width may be too general.