Struct NamedTensor 

pub struct NamedTensor<B: Backend, D: NamedDims<B>> { /* private fields */ }

A tensor with named dimensions.

Implementations

impl<B: Backend, const D: usize, ND> NamedTensor<B, ND>

where ND: NamedDims<B, Tensor = Tensor<B, D>>,

pub fn from_tensor(tensor: Tensor<B, D>) -> Self

Create a named tensor from a tensor.

pub fn random<S: Into<Shape>>( shape: S, distribution: Distribution, device: &B::Device, ) -> Self

Create a random named tensor of the given shape where each element is sampled from the given distribution.

pub fn shape(&self) -> Shape

Returns the shape of the current tensor.

pub fn mul(self, rhs: Self) -> Self

Applies element wise multiplication operation.

y = x2 * x1

pub fn reshape<const D2: usize, S, ND2>( self, shape: S, _: ND2, ) -> NamedTensor<B, ND2>

where S: Into<Shape>, ND2: NamedDims<B, Tensor = Tensor<B, D2>>,

Reshape the tensor to have the given shape.

Panics

If the tensor can not be reshape to the given shape.

impl<B: Backend, const D: usize, ND> NamedTensor<B, ND>

where ND: NamedDims<B, Tensor = Tensor<B, D>>,

pub fn matmul<NamedDimsRhs, NamedDimsOut>( self, rhs: NamedTensor<B, NamedDimsRhs>, ) -> NamedTensor<B, NamedDimsOut>

where NamedDimsRhs: NamedDims<B, Tensor = Tensor<B, D>>, NamedDimsOut: NamedDims<B, Tensor = Tensor<B, D>>, Self: Matmul<NamedTensor<B, NamedDimsRhs>, NamedTensor<B, NamedDimsOut>>,

Applies the matrix multiplication operation.

C = AB

Panics

If the two tensors dont’ have a compatible shape.

impl<B: Backend, const D: usize, ND> NamedTensor<B, ND>

where ND: NamedDims<B, Tensor = Tensor<B, D>>,

pub fn swap_dims<ND2, const D1: usize, const D2: usize>( self, ) -> NamedTensor<B, ND2>

where ND2: NamedDims<B, Tensor = Tensor<B, D>>, Self: SwapDims<NamedTensor<B, ND2>, D1, D2>,

Swap two dimensions.

Trait Implementations

impl<B: Clone + Backend, D: Clone + NamedDims<B>> Clone for NamedTensor<B, D>

where D::Tensor: Clone,

fn clone(&self) -> NamedTensor<B, D>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<B: Debug + Backend, D: Debug + NamedDims<B>> Debug for NamedTensor<B, D>

where D::Tensor: Debug,

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

Formats the value using the given formatter.

impl<B: Backend, const D: usize, ND> Display for NamedTensor<B, ND>

where ND: NamedDims<B, Tensor = Tensor<B, D>> + NamedDims<B>,

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

Formats the value using the given formatter.

impl<B: Backend, ND: NamedDims<B, Tensor = Tensor<B, D>>, const D: usize> From<NamedTensor<B, ND>> for Tensor<B, D>

fn from(nt: NamedTensor<B, ND>) -> Self

Converts to this type from the input type.

impl<B: Backend, ND: NamedDims<B, Tensor = Tensor<B, D>>, const D: usize> From<Tensor<B, D>> for NamedTensor<B, ND>

fn from(tensor: Tensor<B, D>) -> Self

Converts to this type from the input type.