reinforcex 0.0.2

Deep Reinforcement Learning Framework
1
2
3
4
5
6
7

use tch::Tensor;

pub trait BaseQFunction {
    fn forward(&self, x: &Tensor) -> Tensor;
    fn is_cuda(&self) -> bool;
    fn clone(&self) -> Box<dyn BaseQFunction>;
}