Struct QNetwork 

pub struct QNetwork { /* private fields */ }

Q-Network (simple MLP)

Implementations

impl QNetwork

pub fn new(state_dim: usize, action_dim: usize, hidden_dim: usize) -> Self

Create a new Q-Network

pub fn forward(&self, state: &Tensor) -> Tensor

Forward pass: compute Q-values for all actions

pub fn q_value(&self, state: &Tensor, action: usize) -> f32

Get Q-value for a specific action

Trait Implementations

impl Clone for QNetwork

fn clone(&self) -> QNetwork

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for QNetwork

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.