My current input shape is [4, 3, 32, 32] and I want to reduce the second parameter to 1 which will eventually become [4, 1, 32, 32]. Are there any methods that can achieve this