feat(pt): support complete form energy loss#3782

wanghan-iapcm merged 3 commits intodeepmodeling:develdeepmodeling/deepmd-kit:develfrom

iProzd:add_ae_apiProzd/deepmd-kit:add_ae_apCopy head branch name to clipboard