Struct burn_core::tensor::Shape

pub struct Shape<const D: usize> {
    pub dims: [usize; D],
}

Fields

dims: [usize; D]

Implementations

impl<const D: usize> Shape<D>

pub fn new(dims: [usize; D]) -> Shape<D>

Constructs a new Shape.

impl<const D: usize> Shape<D>

pub fn num_elements(&self) -> usize

Trait Implementations

impl<const D: usize> Clone for Shape<D>

fn clone(&self) -> Shape<D>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<const D: usize> Debug for Shape<D>

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

Formats the value using the given formatter.

impl<const D: usize> From<[usize; D]> for Shape<D>

fn from(dims: [usize; D]) -> Shape<D>

Converts to this type from the input type.

impl<const D: usize> From<Vec<i64, Global>> for Shape<D>

fn from(shape: Vec<i64, Global>) -> Shape<D>

Converts to this type from the input type.

impl<const D: usize> From<Vec<u64, Global>> for Shape<D>

fn from(shape: Vec<u64, Global>) -> Shape<D>

Converts to this type from the input type.

impl<const D: usize> From<Vec<usize, Global>> for Shape<D>

fn from(shape: Vec<usize, Global>) -> Shape<D>

Converts to this type from the input type.

impl<const D: usize> PartialEq<Shape<D>> for Shape<D>

fn eq(&self, other: &Shape<D>) -> bool

This method tests for self and other values to be equal, and is used by ==.

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

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

impl<const D: usize> Eq for Shape<D>

impl<const D: usize> StructuralEq for Shape<D>

impl<const D: usize> StructuralPartialEq for Shape<D>