Struct TensorShape 

pub struct TensorShape {
    pub n: usize,
    pub c: usize,
    pub h: usize,
    pub w: usize,
}

Tensor shape for 4D tensors (N, C, H, W).

Fields

n: usize

Batch size

c: usize

Number of channels

h: usize

Height

w: usize

Width

Implementations

impl TensorShape

pub fn new(n: usize, c: usize, h: usize, w: usize) -> Self

Creates a new tensor shape.

pub fn numel(&self) -> usize

Returns the total number of elements.

pub fn spatial_size(&self) -> usize

Returns the spatial size (H * W).

pub fn channel_size(&self) -> usize

Returns the channel size (C * H * W).

Trait Implementations

impl Clone for TensorShape

fn clone(&self) -> TensorShape

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorShape

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

Formats the value using the given formatter.

impl Display for TensorShape

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

Formats the value using the given formatter.

impl PartialEq for TensorShape

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for TensorShape

impl Eq for TensorShape

impl StructuralPartialEq for TensorShape