sbi.inference.vector_field_estimator_based_potential#
- vector_field_estimator_based_potential(vector_field_estimator, prior, x_o, enable_transform=True, **kwargs)[source]#
Returns the potential function gradient for vector field estimators.
- Parameters:
vector_field_estimator (ConditionalVectorFieldEstimator) – The neural network modelling the vector field.
prior (Distribution | None) – The prior distribution.
x_o (Tensor | None) – The observed data at which to evaluate the vector field.
enable_transform (bool) – Whether to enable transforms. Not supported yet.
**kwargs – Additional keyword arguments passed to VectorFieldBasedPotential.
- Returns:
The potential function and a transformation that maps to unconstrained space.
- Return type: