def mnp_stack(*arrs):
a = mnp.stack(arrs, axis=-4)
b = mnp.stack(arrs, axis=-3)
c = mnp.stack(arrs, axis=0)
d = mnp.stack(arrs, axis=3)
e = mnp.stack(arrs, axis=2)
return a, b, c, d, e
def stack(data, axis=0):
return mnp.stack(data, axis)