more WIP on the output layer in the DL NN impl