nn_run2 now uses softmax/cross-entropy
using slices of a 1D array for speed