keras/src/ops/nn.py/CategoricalCrossentropy/__init__
class CategoricalCrossentropy:    def __init__(self, from_logits=False, axis=-1):
        super().__init__()
        self.from_logits = from_logits
        self.axis = axis