Define dtype

Signed-off-by: Beat Buesser <[email protected]>

Author: Beat Buesser

art/preprocessing/standardisation_mean_std/standardisation_mean_std_pytorch.py

@@ -72,8 +72,8 @@ def forward(
         """
         import torch  # lgtm [py/repeated-import]
 
-        mean = torch.tensor(self.mean, device=self._device)
-        std = torch.tensor(self.std, device=self._device)
+        mean = torch.tensor(self.mean, device=self._device, dtype=torch.float32)
+        std = torch.tensor(self.std, device=self._device, dtype=torch.float32)
 
         x_norm = x - mean
         x_norm = x_norm / std