Trait burn_tensor::ReshapeArgs

pub trait ReshapeArgs<const D2: usize> {
    // Required method
    fn into_shape<B: Backend, const D: usize, K: BasicOps<B>>(
        self,
        tensor: &Tensor<B, D, K>,
    ) -> Shape<D2>;
}

Trait used for reshape arguments.

Required Methods

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

Converts to a shape.

Object Safety

This trait is not object safe.

Implementations on Foreign Types

impl<const D2: usize> ReshapeArgs<D2> for [i32; D2]

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

impl<const D2: usize> ReshapeArgs<D2> for [usize; D2]

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

Implementors

impl<const D2: usize> ReshapeArgs<D2> for Shape<D2>