kd.summaries.Histogram

kd.summaries.Histogram#

class kauldron.summaries.Histogram(
*,
tensor: jaxtyping.Shaped[Array, 'n'] | jaxtyping.Shaped[ndarray, 'n'],
num_buckets: int,
)[source]

Bases: object

Output type for histogram summaries.

tensor: jaxtyping.Shaped[Array, 'n'] | jaxtyping.Shaped[ndarray, 'n']
num_buckets: int