disjunction_module.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <optional>
#include "pulsatrix/module.hpp"
 
namespace pulsatrix {
 
class DisjunctionModule : public Module {
public:
    enum class TConorm {
        Product,      
        Lukasiewicz,  
        Godel         
    };
 
    explicit DisjunctionModule(DeviceBackend* backend, TConorm t_conorm = TConorm::Product);
 
    using Module::forward;
 
    [[nodiscard]] Tensor forward(const Tensor& a, const Tensor& b);
 
    [[nodiscard]] static Tensor stack_operands(const Tensor& a, const Tensor& b, DeviceBackend* backend);
 
    [[nodiscard]] Tensor backward(const Tensor& grad_output) override;
 
    [[nodiscard]] OpType op_type() const override { return OpType::Elementwise; }
 
    [[nodiscard]] Tensor propagate_relevance(const Tensor& relevance_out, const LRPRuleConfig& config) override;
 
 
    [[nodiscard]] std::optional<DeviceType> compute_device() const override { return backend_->device(); }
 
protected:
    [[nodiscard]] Tensor forward_impl(const Tensor& input) override;
 
private:
    DeviceBackend* backend_;
    TConorm t_conorm_;
    Tensor last_input_;
    Tensor last_output_;
    bool has_forwarded_ = false;
};
 
}  // namespace pulsatrix