Trait web_rwkv::tensor::TensorExt

pub trait TensorExt: Sized {
    // Required method
    fn shape(&self) -> Shape;

    // Provided method
    fn check_shape(&self, shape: Shape) -> Result<(), TensorError> { ... }
}

Required Methods

fn shape(&self) -> Shape

Provided Methods

fn check_shape(&self, shape: Shape) -> Result<(), TensorError>

Object Safety

This trait is not object safe.

Implementors

impl<T: Scalar> TensorExt for TensorView<'_, T>

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

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