16 lines
No EOL
368 B
Python
16 lines
No EOL
368 B
Python
from ulab import numpy as np
|
|
from .multiarray import asarray
|
|
|
|
def atleast_1d(*arys):
|
|
res = []
|
|
for ary in arys:
|
|
ary = asarray(ary)
|
|
if not isinstance(ary,(np.ndarray)):
|
|
result = ary.reshape((1,))
|
|
else:
|
|
result = ary
|
|
res.append(result)
|
|
if len(res) == 1:
|
|
return res[0]
|
|
else:
|
|
return res |