kd.summaries.HistogramSummary#

class kauldron.summaries.HistogramSummary(*, tensor: typing.Annotated[typing.Any, <object object at 0x76412092fb90>], num_buckets: int = 30)[source]

Bases: kauldron.metrics.base.Metric

Basic histogram summary.

tensor: Annotated[Any, <object object at 0x76412092fb90>]

num_buckets: int = 30

class State( tensor: Array['n'], *, parent: _MetricT = _EMPTY_TYPE.EMPTY, )[source]

Bases: kauldron.metrics.auto_state.AutoState[HistogramSummary]

Collecting state that returns Histograms.

tensor: Array['n']

compute() → kauldron.summaries.histograms.Histogram[source]: Returns the concatenated and flattened values as a Histogram.

merge( other: kauldron.metrics.auto_state._SelfT, ) → kauldron.metrics.auto_state._SelfT[source]: Checks static fields for equality and merges data-fields.

replace(**updates): Returns a new object replacing the specified fields with new values.

get_state( tensor: jaxtyping.Shaped[Array, '*any'] | jaxtyping.Shaped[ndarray, '*any'], ) → kauldron.summaries.histograms.HistogramSummary.State[source]

empty() → kauldron.metrics.base.Metric.State[source]