How to get per atom energy?

[ipcamit]

I have received a DeepMD potential from collaborator, and it yields total energy of a configuration. How can I get per-atom energy in DeepMD (tensorflow backend)? I want to manually separate the partial energy of some particles.
Also can I get partial forces? For example if I have n atoms, can I ask DeepMD to only give me forces on particle i?

[QuantumMisaka]

How can I get per-atom energy in DeepMD (tensorflow backend)?

Refer to #4546

Also can I get partial forces? For example if I have n atoms, can I ask DeepMD to only give me forces on particle i?

You can just select it from all of the results.

[ipcamit]

Thank you for the reply, #4546 indeed solves my first issue.
However, picking the force from the result is not what I meant. Sorry for not clarifying it properly, but I only want the force contributions to and from a particle i. That is, for a three particle system (i,j,k) I want F_ij, F_ik, but not F_jk. Essentially can I manually backpropogate over the atomic energies (w.r.t positions)?