How to connect a loop with a multidimensional array?

Question

Python

Created a multidimensional array:
data2 = w_new
Need to connect with a neural network loop with 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)

How to connect it to the end so that it works?