diff --git a/monai/transforms/signal/array.py b/monai/transforms/signal/array.py index 1a7b87a774..938f42192c 100644 --- a/monai/transforms/signal/array.py +++ b/monai/transforms/signal/array.py @@ -380,7 +380,7 @@ def __call__(self, signal: NdarrayOrTensor) -> NdarrayOrTensor: Args: signal: signal to be filled """ - signal = torch.nan_to_num(convert_to_tensor(signal), nan=self.replacement) + signal = torch.nan_to_num(convert_to_tensor(signal, track_meta=True), nan=self.replacement) return signal