environment.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <cstdint>
 
#include "pulsatrix/tensor.hpp"
 
namespace pulsatrix {
 
struct StepResult {
    Tensor observation;
    float reward;
    bool done;
};
 
class Environment {
public:
    virtual ~Environment() = default;
 
    [[nodiscard]] virtual Tensor reset() = 0;
 
    [[nodiscard]] virtual Tensor reset(const Tensor& initial_state) = 0;
 
    [[nodiscard]] virtual StepResult step(const Tensor& action) = 0;
 
    [[nodiscard]] virtual int64_t observation_dim() const = 0;
 
    [[nodiscard]] virtual int64_t action_dim() const = 0;
 
    [[nodiscard]] virtual bool is_discrete() const = 0;
};
 
}  // namespace pulsatrix