Changes to Tensor labels handling

2025-02-25 12:44:18 +01:00
parent f67467e5bd
commit 8b797d589a
2 changed files with 4 additions and 2 deletions
--- a/pina/model/block/convolution_2d.py
+++ b/pina/model/block/convolution_2d.py
@@ -500,6 +500,7 @@ class ContinuousConvBlock(BaseContinuousConv):

        # initialize grid
        X = self._grid_transpose.clone().detach()
+        
        conv_transposed = self._grid_transpose.clone().detach()

        # list to iterate for calculating nn output