MultipleOutputs.emadl 497 Bytes
Newer Older
1
component MultipleOutputs{
2 3
    ports in Z(0:255)^{3, 32, 32} data,
         out Q(0:1)^{10} pred[2];
4 5 6

    implementation CNN {
        data ->
7 8 9 10 11
        Convolution(kernel=(5,5), channels=8, padding="valid") ->
        Convolution(kernel=(5,5), channels=8, padding="valid") ->
        FullyConnected(units=128) ->
        Dropout() ->
        FullyConnected(units=10) ->
12 13 14 15 16 17
        (
            Softmax() ->
            pred[0]
        |
            Softmax() ->
            pred[1]
18
        );
19 20
    }
}