Skip to content

Commit 8dbf689

Browse files
Move statistical functions to dpctl_ext.tensor and reuse them
1 parent 99c75cb commit 8dbf689

3 files changed

Lines changed: 394 additions & 7 deletions

File tree

dpctl_ext/tensor/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -181,6 +181,7 @@
181181
unique_values,
182182
)
183183
from ._sorting import argsort, sort, top_k
184+
from ._statistical_functions import mean, std, var
184185
from ._type_utils import can_cast, finfo, iinfo, isdtype, result_type
185186

186187
__all__ = [
@@ -267,6 +268,7 @@
267268
"log10",
268269
"max",
269270
"maximum",
271+
"mean",
270272
"meshgrid",
271273
"min",
272274
"minimum",
@@ -308,6 +310,7 @@
308310
"square",
309311
"squeeze",
310312
"stack",
313+
"std",
311314
"subtract",
312315
"sum",
313316
"swapaxes",
@@ -327,6 +330,7 @@
327330
"unique_inverse",
328331
"unique_values",
329332
"unstack",
333+
"var",
330334
"vecdot",
331335
"where",
332336
"zeros",

0 commit comments

Comments
 (0)