Trait web_rwkv::tensor::TensorReshape

pub trait TensorReshape: Sized {
    // Required method
    fn reshape(
        &self,
        x: TensorDimension,
        y: TensorDimension,
        z: TensorDimension,
        w: TensorDimension
    ) -> Result<Self, TensorError>;
}

Required Methods

fn reshape( &self, x: TensorDimension, y: TensorDimension, z: TensorDimension, w: TensorDimension ) -> Result<Self, TensorError>

Object Safety

This trait is not object safe.

Implementors

impl<T: Scalar> TensorReshape for TensorCpu<'_, T>

impl<T: Scalar, K: Kind> TensorReshape for TensorGpu<T, K>