BUG: zeros dtype defaults to float64

Author: ev-br

torch_np/_helpers.py

@@ -94,6 +94,8 @@ def to_tensors_or_none(*inputs):
 
 
 def _outer(x, y):
+    from ._ndarray import asarray
+
     x_tensor, y_tensor = to_tensors(x, y)
     result = torch.outer(x_tensor, y_tensor)
     return asarray(result)

torch_np/_wrapper.py

@@ -7,7 +7,6 @@
 
 import torch
 
-from . import _dtypes, _helpers, _decorators   # isort : skip
 from ._detail import _flips, _reductions, _util
 from ._detail import implementations as _impl
 from ._ndarray import (
@@ -20,6 +19,8 @@
     result_type,
 )
 
+from . import _decorators, _dtypes, _helpers  # isort: skip
+
 # Things to decide on (punt for now)
 #
 # 1. Q: What are the return types of wrapper functions: plain torch.Tensors or
@@ -272,12 +273,15 @@ def ones_like(a, dtype=None, order="K", subok=False, shape=None):
     return result
 
 
-# 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))