probly.representation.sample.torch_functions.torch_function_override

probly.representation.sample.torch_functions.torch_function_override(torch_func: _BoundTorchFunction) _TorchFunction[source]

Decorator to convert a bound torch function to a torch function.