Struct TensorStats 

pub struct TensorStats {
    pub tensor_id: usize,
    pub shape: Vec<usize>,
    pub num_elements: usize,
    pub dtype: String,
    pub min_value: Option<f64>,
    pub max_value: Option<f64>,
    pub mean_value: Option<f64>,
    pub std_dev: Option<f64>,
    pub num_nans: Option<usize>,
    pub num_infs: Option<usize>,
}

Tensor statistics for inspection.

Fields

tensor_id: usize

Tensor ID

shape: Vec<usize>

Shape of the tensor

num_elements: usize

Number of elements

dtype: String

Data type

min_value: Option<f64>

Minimum value (if computed)

max_value: Option<f64>

Maximum value (if computed)

mean_value: Option<f64>

Mean value (if computed)

std_dev: Option<f64>

Standard deviation (if computed)

num_nans: Option<usize>

Number of NaN values

num_infs: Option<usize>

Number of infinite values

Implementations

impl TensorStats

pub fn new( tensor_id: usize, shape: Vec<usize>, dtype: impl Into<String>, ) -> Self

Create tensor stats with basic information.

pub fn with_statistics( self, min: f64, max: f64, mean: f64, std_dev: f64, num_nans: usize, num_infs: usize, ) -> Self

Add computed statistics.

pub fn has_numerical_issues(&self) -> bool

Check if the tensor has numerical issues.

Trait Implementations

impl Clone for TensorStats

fn clone(&self) -> TensorStats

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorStats

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

Formats the value using the given formatter.

impl Display for TensorStats

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

Formats the value using the given formatter.