reinforcex 0.0.4

Deep Reinforcement Learning Framework
1
2
3
4
5
6
7

use crate::prob_distributions::BaseDistribution;
use tch::{Device, Tensor};

pub trait BasePolicy {
    fn forward(&self, x: &Tensor) -> (Box<dyn BaseDistribution>, Option<Tensor>);
    fn device(&self) -> Device;
}