probly.representation.sample.torch_functions.torch_stack_function¶

probly.representation.sample.torch_functions.torch_stack_function(func: Callable, args: tuple[Any, ...], kwargs: dict[str, Any]) → Any[source]¶: Implementation of torch.stack for sample tensors.