Python-сообщество

помогите и объясните пожалуйста как это сделать на python недавно начал работать
многомерный массив который я создал
data2=w_new
и теперь его мне нужно соединить с циклом нейронной сети с LSTM
def f(x,x1):
return x+x1

NN=10
N=5
n=272

data2=np.zeros((NN,N,n))
for j in range(NN):
data2=2*np.random.rand(n)-1
data2=2*np.random.rand(n)-1

for i in range(2,N):
data=f(data,data)

как это соединить до конца, чтобы это работало