Line: 286 https://github.com/deepmodeling/deepmd-kit/blob/76371ceed2fd0d8e259e482e046da350d4006c0a//deepmd/tf/common.py#L283-L293