Struct Shape

pub struct Shape {
    pub dims: Vec<usize>,
}

Shape of a tensor.

Fields

dims: Vec<usize>

The dimensions of the tensor.

Implementations

impl Shape

pub fn num_elements(&self) -> usize

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

pub fn num_dims(&self) -> usize

Returns the number of dimensions.

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

Constructs a new Shape.

pub fn dims<const D: usize>(&self) -> [usize; D]

Returns the dimensions of the tensor as an array.

pub fn flatten(&self) -> Self

Change the shape to one dimensional with the same number of elements.

Trait Implementations

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

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

Converts to a shape.

impl Clone for Shape

fn clone(&self) -> Shape

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Shape

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

Formats the value using the given formatter.

impl From<&Vec<usize>> for Shape

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

Converts to this type from the input type.

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

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

Converts to this type from the input type.

impl From<Shape> for Vec<usize>

fn from(shape: Shape) -> Self

Converts to this type from the input type.

impl From<Vec<i64>> for Shape

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

Converts to this type from the input type.

impl From<Vec<u64>> for Shape

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

Converts to this type from the input type.

impl From<Vec<usize>> for Shape

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

Converts to this type from the input type.

impl PartialEq for Shape

fn eq(&self, other: &Shape) -> 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

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

Converts to a shape.

impl Eq for Shape

impl StructuralPartialEq for Shape