A random neural network
X = torch.randn(1000, 2)
for i in range(5):
model = nn.Sequential(
nn.Linear(2, 5),
nn.ReLU(),
nn.Linear(5, 2)
)
model(X)
X = torch.randn(1000, 2)
for i in range(5):
model = nn.Sequential(
nn.Linear(2, 5),
nn.ReLU(),
nn.Linear(5, 2)
)
model(X)