geom (Geometry) – Geometry object.
a (Array) – batch of histograms of shape [batch, num_a] where num_a matches
the first value of the shape attribute of
geom.
weights (Optional[Array]) – [batch,] positive weights summing to \(1\). Uniform by
default.
