Move definition of get_non_neuron_inputs outside of loop and rename
No num_output needed for pool layer