attribution.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <string>
#include <unordered_map>
 
#include "pulsatrix/device_backend.hpp"
#include "pulsatrix/tensor.hpp"
 
namespace pulsatrix {
 
struct Attribution {
    std::string method;
 
    Tensor values;
 
    std::unordered_map<std::string, std::string> metadata;
};
 
namespace explainer_detail {
 
inline DeviceBackend* backend_beside(const Tensor& like, DeviceBackend* backend) {
    return (backend != nullptr && backend->device() == like.device()) ? backend : like.backend();
}
 
}  // namespace explainer_detail
 
}  // namespace pulsatrix