feat(pt): make se_a and dpa1 compatible with torch.export#5193

wanghan-iapcm wants to merge 3 commits into

deepmodeling:masterdeepmodeling/deepmd-kit:masterfrom

wanghan-iapcm:feat-export-descrptwanghan-iapcm/deepmd-kit:feat-export-descrptCopy head branch name to clipboard