Struct border_tch_agent::TensorSubBatch

pub struct TensorSubBatch { /* private fields */ }

A buffer consisting of a Tensor.

The internal buffer of this struct has the shape of [n_capacity, shape[1..]], where shape is obtained from the data pushed at the first time via TensorSubBatch::push method. [1..] means that the first axis of the given data is ignored as it might be batch size.

Implementations

impl TensorSubBatch

pub fn from_tensor(t: Tensor) -> Self

Trait Implementations

impl Clone for TensorSubBatch

fn clone(&self) -> Self

Returns a copy of the value.

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

Performs copy-assignment from source.

impl From<TensorSubBatch> for Tensor

fn from(b: TensorSubBatch) -> Self

Converts to this type from the input type.

impl SubBatch for TensorSubBatch

fn push(&mut self, index: usize, data: &Self)

Pushes given data.

If the internal buffer is empty, it will be initialized with the shape [capacity, data.buf.size()[1..]].

fn new(capacity: usize) -> Self

Builds a subbatch with a capacity.

fn sample(&self, ixs: &Vec<usize>) -> Self

Takes samples in the batch.