satisfaction_loss.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "pulsatrix/aggregator_module.hpp"
#include "pulsatrix/device_backend.hpp"
#include "pulsatrix/tensor.hpp"
 
namespace pulsatrix {
 
class SatisfactionLoss {
public:
    explicit SatisfactionLoss(DeviceBackend* backend, float p = 2.0f);
 
    [[nodiscard]] float forward(const Tensor& truth_values);
 
    [[nodiscard]] Tensor backward();
 
private:
    DeviceBackend* backend_;
    AggregatorModule aggregator_;
    bool has_forwarded_ = false;
};
 
}  // namespace pulsatrix