keras/src/ops/nn.py/CategoricalCrossentropy/call
class CategoricalCrossentropy:    def call(self, target, output):
        return backend.nn.categorical_crossentropy(
            target, output, from_logits=self.from_logits, axis=self.axis
        )