probly.representation.sample.array_functions.track_sample_axis_after_reduction¶

probly.representation.sample.array_functions.track_sample_axis_after_reduction(original_sample_dim: int, original_ndim: int, axis: int | tuple[int, ...] | None, keepdims: bool) → int | None[source]¶: Track the sample axis after a reduction operation.