How to extract the force field from latent_sde?

[xiniyiwang]

Hi,
Following latent_sde_lorenz.py, I can learn the latent dynamics from my 2D dataset successfully. It performs well:

But what should I do if I want to extract the deterministic force (drift term) in the stochastic equation, and project the drift term to 2D space? Then I can get the phase portrait in 2D space as follows:

Similarly, how can I show the diffusion term explicitly from the learned SDE?

I would appreciate anyone willing to offer me some advice.