INF201/uke5.py

26 lines
402 B
Python
Raw Normal View History

2023-10-25 12:12:34 +00:00
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)