Skip to content

Commit 22d231c

Browse files
Added ones, full, empty_like, zeros_like, ones_like and full_like ctors (#822)
* Added ones, full, empty_like, zeros_like, ones_like and full_like ctors * Fixed error in arange shape computation dpt.arange(4, 0, -1) was producing array of wrong size. Bug fixed, test to be added. * Expanded tests 1. Added test for arange with negative step 2. Added tests for empty_like, zeros_like, ones_like, full_like * Added arg validation tests for some constructors
2 parents 77e6a60 + dc7aa6c commit 22d231c

File tree

5 files changed

+804
-111
lines changed

5 files changed

+804
-111
lines changed

dpctl/tensor/__init__.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,18 @@
2222
"""
2323

2424
from dpctl.tensor._copy_utils import asnumpy, astype, copy, from_numpy, to_numpy
25-
from dpctl.tensor._ctors import arange, asarray, empty, zeros
25+
from dpctl.tensor._ctors import (
26+
arange,
27+
asarray,
28+
empty,
29+
empty_like,
30+
full,
31+
full_like,
32+
ones,
33+
ones_like,
34+
zeros,
35+
zeros_like,
36+
)
2637
from dpctl.tensor._device import Device
2738
from dpctl.tensor._dlpack import from_dlpack
2839
from dpctl.tensor._manipulation_functions import (
@@ -46,6 +57,12 @@
4657
"copy",
4758
"empty",
4859
"zeros",
60+
"ones",
61+
"full",
62+
"empty_like",
63+
"zeros_like",
64+
"ones_like",
65+
"full_like",
4966
"flip",
5067
"reshape",
5168
"roll",

0 commit comments

Comments
 (0)