Struct TensorBatch 

pub struct TensorBatch { /* 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 TensorBatch::push method. [1..] means that the first axis of the given data is ignored as it might be batch size.

Implementations

impl TensorBatch

pub fn from_tensor(t: Tensor) -> Self

Trait Implementations

impl BatchBase for TensorBatch

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

Creates a new batch with the specified capacity.

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

Retrieves samples from the specified indices.

impl Clone for TensorBatch

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl From<TensorBatch> for Tensor

fn from(b: TensorBatch) -> Self

Converts to this type from the input type.