Struct TensorBatch 

pub struct TensorBatch { /* private fields */ }

A buffer consisting of a Tensor.

The internal buffer is Vec<Tensor>.

Implementations

impl TensorBatch

pub fn from_tensor(t: Tensor) -> Self

pub fn to(&mut self, device: &Device) -> Result<()>

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.dims()[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) -> TensorBatch

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorBatch

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

Formats the value using the given formatter.

impl From<TensorBatch> for Tensor

fn from(b: TensorBatch) -> Self

Converts to this type from the input type.