keras/src/ops/math.py/Logsumexp/compute_output_spec
class Logsumexp:    def compute_output_spec(self, x):
        output_shape = reduce_shape(x.shape, self.axis, self.keepdims)
        return KerasTensor(shape=output_shape)