Invariant.emadl 332 Bytes
Newer Older
Sebastian N.'s avatar
Sebastian N. committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
component Invariant{
    ports in Z(0:3)^{1} data[2],
         out Q(0:1)^{4} pred[3];

    implementation CNN {
        data[0] ->
        FullyConnected(units=4) ->
        Softmax() ->
        pred[0];

        data[1] ->
        OneHot(size=4) ->
        pred[1];

        1 ->
        OneHot(size=4) ->
        pred[2];
    }
}