array-api-extra/src/array_api_extra/__init__.py at 33c9ad8e891f3fbd78652fa3e107b570ca490b0d · data-apis/array-api-extra · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
"""Extra array functions built on top of the array API standard."""

from ._delegation import (
    argpartition,
    atleast_nd,
    cov,
    expand_dims,
    fill_diagonal,
    isclose,
    isin,
    nan_to_num,
    one_hot,
    pad,
    partition,
    setdiff1d,
    sinc,
)
from ._lib._at import at
from ._lib._funcs import (
    apply_where,
    broadcast_shapes,
    create_diagonal,
    default_dtype,
    kron,
    nunique,
)
from ._lib._lazy import lazy_apply

__version__ = "0.9.1.dev0"

# pylint: disable=duplicate-code
__all__ = [
    "__version__",
    "apply_where",
    "argpartition",
    "at",
    "atleast_nd",
    "broadcast_shapes",
    "cov",
    "create_diagonal",
    "default_dtype",
    "expand_dims",
    "fill_diagonal",
    "isclose",
    "isin",
    "kron",
    "lazy_apply",
    "nan_to_num",
    "nunique",
    "one_hot",
    "pad",
    "partition",
    "setdiff1d",
    "sinc",
]