probly.representation.sample.torch_functions.torch_internals_override¶

probly.representation.sample.torch_functions.torch_internals_override(torch_sample_param_name: str) → Callable[[_BoundTorchFunctionWithInternals], _TorchFunction][source]¶
probly.representation.sample.torch_functions.torch_internals_override(*, torch_sample_param_pos: int) → Callable[[_BoundTorchFunctionWithInternals], _TorchFunction]: Decorator to convert a function taking a sample tensor argument.