BUG: zeros dtype defaults to float64

Author: ev-br

torch_np/_wrapper.py

@@ -274,10 +274,13 @@ def ones_like(a, dtype=None, order="K", subok=False, shape=None):
 
 # XXX: dtype=float
 @_decorators.dtype_to_torch
-def zeros(shape, dtype=float, order="C", *, like=None):
+def zeros(shape, dtype=None, order="C", *, like=None):
     _util.subok_not_ok(like)
     if order != "C":
         raise NotImplementedError
+    if dtype is None:
+        from ._detail._scalar_types import default_float_type
+        dtype = default_float_type.torch_dtype
     return asarray(torch.zeros(shape, dtype=dtype))