summaryrefslogtreecommitdiff
path: root/numpy/_array_api/_manipulation_functions.py
blob: 19e9c1cab45a93c4a26caa3b9572cd4c70a8ca7f (plain)
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
def concat(arrays, /, *, axis=0):
    # Note: the function name is different here
    from .. import concatenate
    return concatenate(arrays, axis=axis)

def expand_dims(x, axis, /):
    from .. import expand_dims
    return expand_dims(x, axis)

def flip(x, /, *, axis=None):
    from .. import flip
    return flip(x, axis=axis)

def reshape(x, shape, /):
    from .. import reshape
    return reshape(x, shape)

def roll(x, shift, /, *, axis=None):
    from .. import roll
    return roll(x, shift, axis=axis)

def squeeze(x, /, *, axis=None):
    from .. import squeeze
    return squeeze(x, axis=axis)

def stack(arrays, /, *, axis=0):
    from .. import stack
    return stack(arrays, axis=axis)