import torch


import utils


net1 = utils.SimpleMLP([28*28,100,50,1])
net2 = utils.SimpleMLP([1,50,100,28*28])


test = torch.randn(10,28*28)
result1 = net1(test) #10*1
result2 = net2(result1) #10*28*28

assert result1.shape[0] == 10
assert result2.shape[0] == 10

assert result1.shape[1] == 1
assert result2.shape[1] == 28*28

7385ea1f8a62356f7e0e55b9f927b728ca50d7d1
<mask token>

<mask token>
assert result1.shape[0] == 10
assert result2.shape[0] == 10
assert result1.shape[1] == 1
assert result2.shape[1] == 28 * 28

<mask token>
net1 = utils.SimpleMLP([28 * 28, 100, 50, 1])
net2 = utils.SimpleMLP([1, 50, 100, 28 * 28])
test = torch.randn(10, 28 * 28)
result1 = net1(test)
result2 = net2(result1)
assert result1.shape[0] == 10
assert result2.shape[0] == 10
assert result1.shape[1] == 1
assert result2.shape[1] == 28 * 28

import torch
import utils
net1 = utils.SimpleMLP([28 * 28, 100, 50, 1])
net2 = utils.SimpleMLP([1, 50, 100, 28 * 28])
test = torch.randn(10, 28 * 28)
result1 = net1(test)
result2 = net2(result1)
assert result1.shape[0] == 10
assert result2.shape[0] == 10
assert result1.shape[1] == 1
assert result2.shape[1] == 28 * 28

import torch


import utils


net1 = utils.SimpleMLP([28*28,100,50,1])
net2 = utils.SimpleMLP([1,50,100,28*28])


test = torch.randn(10,28*28)
result1 = net1(test) #10*1
result2 = net2(result1) #10*28*28

assert result1.shape[0] == 10
assert result2.shape[0] == 10

assert result1.shape[1] == 1
assert result2.shape[1] == 28*28
