Enum InputSpec 

pub enum InputSpec {
    Vector {
        size: usize,
    },
    Grid {
        channels: usize,
        height: usize,
        width: usize,
    },
}

Input specification for the model

Variants

Vector

1D vector input (e.g., for CartPole, SimpleBandit)

Fields

size: usize

Length of the flat input vector. Callers must pass exactly this many floats into UniversalModel::forward.

Grid

3D grid input (e.g., for Snake)

Fields

channels: usize

Number of input channels stacked along the leading axis (e.g. 5 for Snake’s body / head / food / etc. planes).

height: usize

Grid height in cells.

width: usize

Grid width in cells. The expected flat input length is channels * height * width, laid out as channel-major, then row-major within each channel.

Trait Implementations

impl Clone for InputSpec

fn clone(&self) -> InputSpec

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for InputSpec

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for InputSpec

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for InputSpec

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.