| 19 | } |
| 20 | |
| 21 | torch::Tensor plainCNN::forward(torch::Tensor x){ |
| 22 | x = conv1->forward(x); |
| 23 | x = down1->forward(x); |
| 24 | x = conv2->forward(x); |
| 25 | x = down2->forward(x); |
| 26 | x = conv3->forward(x); |
| 27 | x = down3->forward(x); |
| 28 | x = out_conv->forward(x); |
| 29 | return x; |
| 30 | } |