import numpy as np

def relu(y):
    return np.maximum(0, y)

def layer(W, x, b):
    return relu(W @ x + b)

n = [64,128,128,128,10]

print(f"Dimensions: {n}")

# First layer
x = np.random.rand(n[0])
b = np.random.rand(n[1])
y = np.random.rand(128,64)

y = layer(y, x, b)

for i in range(2, len(n)):
    W = np.random.rand(n[i], n[i - 1])
    b = np.random.rand(n[i])
    y = layer(W, y, b)

print(y)