Just curious, that 4 * kernel_size * kernel_size * 16 works only when the kernel size is 3.
It’s not apparent to me how I can avoid using if statements before those two lines of code, if kernel_size = 3, add 4.