Method MergeBlockImpl

src/architectures/FPNDecoder.cpp:62–67

}

MergeBlockImpl::MergeBlockImpl(std::string policy){
    if(policy!=policies[0] && policy!=policies[1]){
        std::cout<< "`merge_policy` must be one of: ['add', 'cat'], got "+policy;
    }
    _policy = policy;
}

torch::Tensor MergeBlockImpl::forward(std::vector<torch::Tensor> x){
    if(_policy=="add") return sumTensor(x);