metrics_sink.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <string>
 
#include "pulsatrix/tensor.hpp"
 
namespace pulsatrix {
 
class MetricsSink {
public:
    virtual ~MetricsSink() = default;
 
    virtual void log_scalar(const std::string& tag, double value, int step) = 0;
 
    virtual void log_histogram(const std::string& tag, const Tensor& values, int step) = 0;
};
 
class NoOpMetricsSink : public MetricsSink {
public:
    void log_scalar(const std::string&, double, int) override {}
    void log_histogram(const std::string&, const Tensor&, int) override {}
};
 
}  // namespace pulsatrix