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.