feat(pt): add dynamic_index version1

Author: iProzd

deepmd/dpmodel/descriptor/dpa3.py

@@ -32,6 +32,7 @@ def __init__(
         angle_init_use_sin: bool = False,
         smooth_edge_update: bool = False,
         angle_multi_freq: Optional[str] = None,
+        no_sel: bool = False,
     ) -> None:
         r"""The constructor for the RepFlowArgs class which defines the parameters of the repflow block in DPA3 descriptor.
 
@@ -113,6 +114,7 @@ def __init__(
         self.angle_init_use_sin = angle_init_use_sin
         self.smooth_edge_update = smooth_edge_update
         self.angle_multi_freq = angle_multi_freq
+        self.no_sel = no_sel
 
     def __getitem__(self, key):
         if hasattr(self, key):

deepmd/pt/model/descriptor/dpa3.py

@@ -167,6 +167,7 @@ def init_subclass_params(sub_data, sub_class):
             angle_init_use_sin=self.repflow_args.angle_init_use_sin,
             smooth_edge_update=self.repflow_args.smooth_edge_update,
             angle_multi_freq=self.repflow_args.angle_multi_freq,
+            no_sel=self.repflow_args.no_sel,
             exclude_types=exclude_types,
             env_protection=env_protection,
             precision=precision,