probly.representation.sample.torch_functions.torch_reduction_function¶

probly.representation.sample.torch_functions.torch_reduction_function(func: Callable, args: tuple[Any, ...], kwargs: dict[str, Any]) → Any[source]¶: Implementation of dimension-reducing torch functions with a keepdim kwarg.