Struct HostTensor 

pub struct HostTensor {
    pub shape: Vec<usize>,
    pub values: Values,
}

A tensor in host memory: full shape (batch first) and row-major elements.

What TensorModel::run takes and returns. furnace-burn turns one into a Burn tensor and back (its HostTensorExt).

Fields

shape: Vec<usize>

values: Values

Implementations

impl HostTensor

pub fn batch(&self) -> usize

Items along the batch dimension; 0 for a tensor without one (an empty shape), which no model accepts

pub fn slice_batch(&self, start: usize, end: usize) -> Self

Items start..end along the batch dimension

pub fn concat_batch(parts: Vec<Self>) -> Result<Self>

Join tensors of the same type and per-item shape along the batch dimension

Trait Implementations

impl Clone for HostTensor

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for HostTensor

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

Formats the value using the given formatter.

impl PartialEq for HostTensor

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for HostTensor