sequential_module.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <vector>
 
#include "pulsatrix/module.hpp"
 
namespace pulsatrix {
 
class SequentialModule : public Module {
public:
    explicit SequentialModule(std::vector<Module*> layers);
 
    [[nodiscard]] Tensor backward(const Tensor& grad_output) override;
 
    [[nodiscard]] OpType op_type() const override { return OpType::Composite; }
 
    [[nodiscard]] Tensor propagate_relevance(const Tensor& relevance_out, const LRPRuleConfig& config) override;
 
    [[nodiscard]] bool supports_lrp_rule(LRPRule rule) const override {
        for (const Module* layer : layers_) {
            if (!layer->supports_lrp_rule(rule)) {
                return false;
            }
        }
        return true;
    }
 
    [[nodiscard]] std::vector<NamedParamRef> named_parameters() override;
 
    void set_training(bool training) override;
 
    [[nodiscard]] const std::vector<Module*>& layers() const { return layers_; }
 
protected:
    [[nodiscard]] Tensor forward_impl(const Tensor& input) override;
 
private:
    std::vector<Module*> layers_;
    bool has_forwarded_ = false;
};
 
}  // namespace pulsatrix