add in fn to get the final layer activations of a network