# (input)--> [Linear->Sigmoid] -> [Linear->Sigmoid] -->(output)
# handle 10,000 vectors at a time
Z1 = nnet.LinearLayer(input_shape=(rows * cols, nnet_batch), n_out=80)
-A1 = nnet.SigmoidLayer(Z1.Z.shape)
-ZO = nnet.LinearLayer(input_shape=A1.A.shape, n_out=10)
-AO = nnet.SigmoidLayer(ZO.Z.shape)
+A1 = nnet.SigmoidLayer(Z1.shape)
+ZO = nnet.LinearLayer(input_shape=A1.shape, n_out=10)
+AO = nnet.SigmoidLayer(ZO.shape)
net = (Z1, A1, ZO, AO)
res = nnet.forward(net, test_images[:, 0:10000])