probly.representation.sample.torch_functions.track_sample_dim_after_reduction¶

probly.representation.sample.torch_functions.track_sample_dim_after_reduction(original_sample_dim: int, original_ndim: int, dim: int | tuple[int, ...] | list[int] | Size | None, keepdim: bool) → int | None[source]¶: Track the sample dimension after a reduction operation.