Struct relearn::torch::history::features::PackedHistoryFeatures

pub struct PackedHistoryFeatures {
    pub discount_factor: f64,
    pub episode_ranges: Vec<Range<usize>>,
    pub batch_sizes: Option<Vec<i64>>,
    pub batch_sizes_tensor: Option<Tensor>,
    pub observation_features: Option<Tensor>,
    pub actions: Option<Tensor>,
    pub returns: Option<Tensor>,
    pub rewards: Option<Tensor>,
    pub device: Device,
}

Packed history features.

Panics

The PackedHistoryFeaturesView this provides will panic if the requested features is not available.

Fields

discount_factor: f64

episode_ranges: Vec<Range<usize>>

batch_sizes: Option<Vec<i64>>

batch_sizes_tensor: Option<Tensor>

observation_features: Option<Tensor>

actions: Option<Tensor>

returns: Option<Tensor>

rewards: Option<Tensor>

device: Device

Trait Implementations

impl Debug for PackedHistoryFeatures

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

Formats the value using the given formatter.

impl PackedHistoryFeaturesView for PackedHistoryFeatures

fn rewards(&self) -> &Tensor

Packed rewards. A 1D f32 tensor.

fn discount_factor(&self) -> f64

Discount factor for calculating returns.

fn episode_ranges(&self) -> &[Range<usize>]

Episode index ranges sorted in decreasing order of episode length.

fn batch_sizes(&self) -> &[i64]

Batch sizes in the packing.

fn batch_sizes_tensor(&self) -> &Tensor

Batch sizes in the packing. A 1D i64 tensor.

fn observation_features(&self) -> &Tensor

Packed observation features. A 2D f64 tensor.

fn actions(&self) -> &Tensor

Packed action values.

fn returns(&self) -> &Tensor

Packed returns (discounted reward-to-go). A 1D f32 tensor.

fn device(&self) -> Device

Device on which tensors will be placed.

impl PartialEq<PackedHistoryFeatures> for PackedHistoryFeatures

fn eq(&self, other: &PackedHistoryFeatures) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &PackedHistoryFeatures) -> bool

This method tests for !=.

impl StructuralPartialEq for PackedHistoryFeatures