tweaks to layers in BaseNeuralNetworkVectorized