Source code for lacuna.array_api.manipulation.__init__

from .. import _dispatch as _dp
from .. import _namespace as _ns




[docs]
def reshape(x, newshape):
    return _dp.reshape(x, newshape)






[docs]
def squeeze(x, axis=None):
    return getattr(_ns, "squeeze")(x, axis=axis)






[docs]
def expand_dims(x, axis):
    return getattr(_ns, "expand_dims")(x, axis)






[docs]
def moveaxis(x, source, destination):
    return getattr(_ns, "moveaxis")(x, source, destination)






[docs]
def permute_dims(x, axes):
    return _dp.permute_dims(x, axes)






[docs]
def stack(arrays, axis=0):
    return getattr(_ns, "stack")(arrays, axis=axis)






[docs]
def concat(arrays, axis=0):
    return getattr(_ns, "concat")(arrays, axis=axis)






[docs]
def unstack(x, axis=0):
    return getattr(_ns, "unstack")(x, axis=axis)






[docs]
def broadcast_to(x, shape):
    return getattr(_ns, "broadcast_to")(x, shape)






[docs]
def broadcast_arrays(*arrays):
    return getattr(_ns, "broadcast_arrays")(*arrays)






[docs]
def flip(x, axis=None):
    return getattr(_ns, "flip")(x, axis=axis)






[docs]
def roll(x, shift, axis=None):
    return getattr(_ns, "roll")(x, shift, axis=axis)






[docs]
def repeat(x, repeats, axis=None):
    return getattr(_ns, "repeat")(x, repeats, axis=axis)






[docs]
def tile(x, reps):
    return getattr(_ns, "tile")(x, reps)