Skip to content

Commit 00f25b1

Browse files
committed
MAINT: simplify full/dtype
1 parent 49959e1 commit 00f25b1

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

torch_np/_wrapper.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -303,17 +303,16 @@ def empty_like(prototype, dtype=None, order="K", subok=False, shape=None):
303303
result = result.reshape(shape)
304304
return result
305305

306-
306+
@_decorators.dtype_to_torch
307307
def full(shape, fill_value, dtype=None, order="C", *, like=None):
308308
_util.subok_not_ok(like)
309309
if order != "C":
310310
raise NotImplementedError
311311
fill_value = asarray(fill_value).get()
312312
if dtype is None:
313-
torch_dtype = asarray(fill_value).get().dtype
314-
else:
315-
torch_dtype = _dtypes.torch_dtype_from(dtype)
316-
return asarray(torch.full(shape, fill_value, dtype=torch_dtype))
313+
dtype = fill_value.dtype
314+
result = torch.full(shape, fill_value, dtype=dtype)
315+
return asarray(result)
317316

318317

319318
@asarray_replacer()

0 commit comments

Comments
 (0)