Method constructWeighedSumLayer

src/engine/InputQuery.cpp:879–1000

}

bool InputQuery::constructWeighedSumLayer( NLR::NetworkLevelReasoner *nlr,
                                           Map<unsigned, unsigned> &handledVariableToLayer,
                                           unsigned newLayerIndex,
                                           Set<unsigned> &handledEquations )
{
    INPUT_QUERY_LOG( "Attempting to construct weightedSumLayer..." );
    struct NeuronInformation
    {
    public:
        NeuronInformation( unsigned variable, unsigned neuron, const Equation *eq )
            : _variable( variable )
            , _neuron( neuron )
            , _eq( eq )
        {
        }

        NeuronInformation()
            : _eq( NULL )
        {
        }

        unsigned _variable;
        unsigned _neuron;
        const Equation *_eq;
    };

    List<NeuronInformation> newNeurons;

    // Look for equations where all variables except one have already been handled
    const List<Equation> &equations = getEquations();
    unsigned index = 0;
    for ( const auto &eq : equations )
    {
        if ( handledEquations.exists( index++ ) )
            continue;

        // Only consider equalities
        if ( eq._type != Equation::EQ )
            continue;

        List<unsigned> eqVariables = eq.getListParticipatingVariables();
        auto it = eqVariables.begin();
        while ( it != eqVariables.end() )
        {
            if ( handledVariableToLayer.exists( *it ) )
                it = eqVariables.erase( it );
            else
                ++it;
        }

        if ( eqVariables.size() == 1 )
        {
            // Add the surviving variable to the new layer
            newNeurons.append( NeuronInformation( *eqVariables.begin(), newNeurons.size(), &eq ) );
            handledEquations.insert( index - 1 );
        }
    }