Struct web_rwkv::tensor::TensorStack

pub struct TensorStack<'a, T: Scalar> {
    pub tensor: TensorCpu<'a, T>,
    pub cursors: Vec<Cursor>,
}

Stack a batch of tensors of shape [C, T, 1] to one with shape [C, A, 1], with cursors information.

Fields

tensor: TensorCpu<'a, T>

cursors: Vec<Cursor>

Implementations

impl<'a, T: Scalar> TensorStack<'a, T>

pub fn max_batch(&self) -> usize

Number of input batches.

pub fn num_batch(&self) -> usize

Number of non-empty input batches.

pub fn num_token(&self) -> usize

Trait Implementations

impl<T: Scalar> TryFrom<Vec<Tensor<Cpu<'_, T>, T>>> for TensorStack<'_, T>

type Error = TensorError

The type returned in the event of a conversion error.

fn try_from(value: Vec<TensorCpu<'_, T>>) -> Result<Self, Self::Error>

Performs the conversion.