I am using TensorFlow in Python 3 to create a CNN that creates a multi class (i.e. expected output is 3 probabilities out of 92) based on a vector of photon energies shape(2