fix typo

deepmodeling · Jan 14, 2022 · a4c32c7 · a4c32c7
1 parent 4f95d01
commit a4c32c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/fit/ener.py b/deepmd/fit/ener.py
@@ -406,7 +406,7 @@ def build (self,
                 # like inputs, but we don't want to add a dependency on inputs
                 inputs_zero = tf.zeros((nframes, self.dim_descrpt * natoms[0]), dtype=self.fitting_precision)
             else:
-                inputs_zero = tf.zeros((nframes, self.dim_descrpt * natoms[0]), dtype=self.fitting_precision)
+                inputs_zero = tf.zeros_like(inputs, dtype=self.fitting_precision)
 
 
         if bias_atom_e is not None :