Struct TensorMemory 

pub struct TensorMemory {
    pub tensor_idx: usize,
    pub shape: Vec<usize>,
    pub element_count: usize,
    pub bytes: usize,
}

Memory usage estimate for a tensor

Fields

tensor_idx: usize

shape: Vec<usize>

element_count: usize

bytes: usize

Implementations

impl TensorMemory

pub fn new(tensor_idx: usize, shape: Vec<usize>, dtype: DType) -> Self

pub fn megabytes(&self) -> f64

Trait Implementations

impl Clone for TensorMemory

fn clone(&self) -> TensorMemory

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorMemory

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

Formats the value using the given formatter.