flax

d67ed87c - Correct incorrect shape note in ensembling.rst

Correct incorrect shape note in ensembling.rst ensembling.rst mentions a stacked shape batch_metrics_np['loss'] but doesn't correctly call out the post-stacked shape.