I\'m trying to extract the values of the logits of a trained model when you have the softmax activation inside the dense layer.
Something along these lines:
