baybe.transformations.utils.convert_transformation

baybe.transformations.utils.convert_transformation(x: Transformation | TensorCallable, /)[source]

Autowrap a torch callable as transformation (with transformation passthrough).

Return type:

Transformation