Hi,
Softmax is essentially normalising the values in the range [0, 1] The input shape and output shape would be the same if you are using a softmax function. For reference you can check the documentation and implementation of sotmax here 1