Fix PyTorch quantile

tum-pbs · Feb 1, 2024 · 8e0f661 · 8e0f661
1 parent b0f4a7c
commit 8e0f661
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/phiml/backend/torch/_torch_backend.py b/phiml/backend/torch/_torch_backend.py
@@ -398,6 +398,7 @@ def all(self, boolean_tensor, axis=None, keepdims=False):
 
     def quantile(self, x, quantiles):
         x = self.to_float(x)
+        quantiles = self.to_float(quantiles)
         result = torch.quantile(x, quantiles, dim=-1)
         return result