probly.representation.sample.torch_functions.torch_permute_function¶

probly.representation.sample.torch_functions.torch_permute_function(func: Callable, args: tuple[Any, ...], kwargs: dict[str, Any], create_sample: TorchSampleCreator, tensor: TorchLike, sample_dim: int, weights: torch.Tensor | None) → Any[source]¶: Implementation of torch.permute for sample tensors.