subtb_loss.hpp

Go to the documentation of this file.

 
#pragma once
 
namespace pulsatrix {
 
class SubTBLoss {
public:
    SubTBLoss() = default;
 
    [[nodiscard]] float forward(float log_flow_i, float sum_log_pf, float log_flow_j, float sum_log_pb,
                                 float pair_weight_ratio);
 
    [[nodiscard]] float delta() const;
 
    [[nodiscard]] float grad_log_flow_i() const;
 
    [[nodiscard]] float grad_log_flow_j() const;
 
    [[nodiscard]] float grad_weight_for_log_pf_range() const;
 
private:
    float last_delta_ = 0.0f;
    float last_pair_weight_ratio_ = 0.0f;
    bool has_forwarded_ = false;
 
    void require_forwarded() const;
};
 
}  // namespace pulsatrix