Struct burn_tensor::Shape

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

Shape of a tensor.

Fields

dims: [usize; D]

The dimensions of the tensor.

Implementations

impl<const D: usize> Shape<D>

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

Constructs a new Shape.

impl<const D: usize> Shape<D>

pub fn num_elements(&self) -> usize

Returns the total number of elements of a tensor having this shape

Trait Implementations

impl<const D1: usize, const D2: usize> BroadcastArgs<D1, D2> for Shape<D2>

fn into_shape(self, _shape: &Shape<D1>) -> Shape<D2>

Converts to a shape.

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

Formats the value using the given formatter.

impl<const D: usize> From<&Vec<usize>> for Shape<D>

fn from(shape: &Vec<usize>) -> Self

Converts to this type from the input type.

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

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

Converts to this type from the input type.

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

fn from(shape: Shape<D>) -> Self

Converts to this type from the input type.

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

fn from(shape: Vec<i64>) -> Self

Converts to this type from the input type.

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

fn from(shape: Vec<u64>) -> Self

Converts to this type from the input type.

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

fn from(shape: Vec<usize>) -> Self

Converts to this type from the input type.

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

fn eq(&self, other: &Shape<D>) -> 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<const D2: usize> ReshapeArgs<D2> for Shape<D2>

fn into_shape<B: Backend, const D: usize, K: BasicOps<B>>( self, tensor: &Tensor<B, D, K>, ) -> Shape<D2>

Converts to a shape.

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

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