def saturateElements(x):
    x = np.asarray(x)
try:
    x[x > 1] = 1
except TypeError:
    x = min(x, 1)
return x
как я могу создать функцию numpy, которая принимает массив numpy, итерируемый или скалярный?