keras/src/layers/activations/softmax.py/Softmax/__init__
class Softmax:    def __init__(self, axis=-1, **kwargs):
        super().__init__(**kwargs)
        self.supports_masking = True
        self.axis = axis
        self.built = True