Merge branch 'lanl:development' into development

shinkle-lanl · Jan 22, 2024 · a630f14 · a630f14
2 parents 2423590 + 8e99f0e
commit a630f14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hippynn/layers/physics.py b/hippynn/layers/physics.py
@@ -262,7 +262,7 @@ def forward(self, features, species):
 
 class VecMag(torch.nn.Module):
     def forward(self, vector_feature):
-        return torch.norm(vector_feature, dim=1).unsqueeze(1)
+        return torch.norm(vector_feature, dim=1)
 
 
 class CombineEnergy(torch.nn.Module):