JaxSampleInternals¶

class probly.representation.sample.jax_functions.JaxSampleInternals(create: JaxSampleCreator[D], array: D, sample_axis: int, weights: jax.Array | None = None)[source]¶

Bases: Generic

Internal information about a sample array.

array: D¶

create: JaxSampleCreator[D]¶

sample_axis: int¶

weights: jax.Array | None¶