Method constructSoftmaxLayer

src/engine/InputQuery.cpp:1825–1967

}

bool InputQuery::constructSoftmaxLayer( NLR::NetworkLevelReasoner *nlr,
                                        Map<unsigned, unsigned> &handledVariableToLayer,
                                        unsigned newLayerIndex,
                                        Set<NonlinearConstraint *> &handledNLConstraints )
{
    INPUT_QUERY_LOG( "Attempting to construct SoftmaxLayer..." );
    struct NeuronInformation
    {
    public:
        NeuronInformation( unsigned variable,
                           unsigned neuron,
                           const Vector<unsigned> &sourceVariables )
            : _variable( variable )
            , _neuron( neuron )
            , _sourceVariables( sourceVariables )
        {
        }

        unsigned _variable;
        unsigned _neuron;
        Vector<unsigned> _sourceVariables;
    };

    List<NeuronInformation> newNeurons;

    // Look for Softmaxes where all the element variables have already been handled
    const List<NonlinearConstraint *> &nlConstraints = getNonlinearConstraints();

    unsigned currentSourceLayer = 0;
    for ( const auto &nlc : nlConstraints )
    {
        if ( handledNLConstraints.exists( nlc ) )
            continue;

        // Only consider Softmax
        if ( nlc->getType() != SOFTMAX )
            continue;

        const SoftmaxConstraint *softmax = (const SoftmaxConstraint *)nlc;

        // Have all input variables been handled?
        bool missingInput = false;
        bool sourceLayerDiffers = false;
        for ( const auto &input : softmax->getInputs() )
        {
            if ( !handledVariableToLayer.exists( input ) )
            {
                missingInput = true;
                break;
            }
            else if ( _ensureSameSourceLayerInNLR && newNeurons.size() &&
                      handledVariableToLayer[input] != currentSourceLayer )
            {
                sourceLayerDiffers = true;
                break;
            }
        }