kd.metrics.TreeReduce

class kauldron.metrics.TreeReduce(*, metric: kauldron.metrics.base.Metric)

Bases: kauldron.metrics.base._TreeMetric

Applies a metric to a pytree and returns the aggregated result.

The given metric defines the aggregation method.

get_state(

**kwargs,

) → kauldron.metrics.base_state.State

empty() → kauldron.metrics.base_state.State