From 9c3f94d3ec43e101d5e113b26c6698998f508b9b Mon Sep 17 00:00:00 2001
From: Dario Coscia <93731561+dario-coscia@users.noreply.github.com>
Date: Thu, 21 Jul 2022 16:12:21 +0200
Subject: [PATCH] Update pinn.py

---
 pina/pinn.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pina/pinn.py b/pina/pinn.py
index 8a2527b..03929cb 100644
--- a/pina/pinn.py
+++ b/pina/pinn.py
@@ -77,9 +77,9 @@ class PINN(object):
         :param vec torch.tensor: the tensor
         """
         if isinstance(self.error_norm, int):
-            return torch.sum(torch.abs(vec**self.error_norm))**(1./self.error_norm)
+            return torch.linalg.vector_norm(vec, ord = self.error_norm,  dtype=self.dytpe)
         elif self.error_norm == 'mse':
-            return torch.mean(vec**2)
+            return torch.mean(vec.pow(2))
         elif self.error_norm == 'me':
             return torch.mean(torch.abs(vec))
         else: