changed return type of norms.l0 to same as input like lx (was always …

…int)
jonasrauber · Feb 13, 2020 · 6bee896 · 6bee896
1 parent b945b6c
commit 6bee896
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eagerpy/norms.py b/eagerpy/norms.py
@@ -8,7 +8,7 @@
 def l0(
     x: TensorType, axis: Optional[AxisAxes] = None, keepdims: bool = False
 ) -> TensorType:
-    return (x != 0).sum(axis=axis, keepdims=keepdims)
+    return (x != 0).sum(axis=axis, keepdims=keepdims).astype(x.dtype)
 
 
 def l1(