From 8b797d589a587d59a55f25cbd5d7874ede4097b6 Mon Sep 17 00:00:00 2001 From: Matteo Bertocchi Date: Tue, 25 Feb 2025 12:44:18 +0100 Subject: [PATCH] Changes to Tensor labels handling --- pina/label_tensor.py | 5 +++-- pina/model/block/convolution_2d.py | 1 + 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/pina/label_tensor.py b/pina/label_tensor.py index cce141c..d94943b 100644 --- a/pina/label_tensor.py +++ b/pina/label_tensor.py @@ -479,9 +479,10 @@ class LabelTensor(torch.Tensor): # Retrieve selected tensor and labels selected_tensor = super().__getitem__(index) - if not hasattr(self, "_labels"): + if hasattr(self, "_labels"): + original_labels=self._labels + else: return selected_tensor - original_labels = self._labels updated_labels = copy(original_labels) diff --git a/pina/model/block/convolution_2d.py b/pina/model/block/convolution_2d.py index 68df175..61f594e 100644 --- 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