Skip to main content

Mamba2SsdInput

burn_mamba::mamba2::ssd::ssd_path

Struct Mamba2SsdInput

pub struct Mamba2SsdInput<B: Backend> {
    pub x_bnlhp: Tensor<B, 5>,
    pub dt_bnlh: Tensor<B, 4>,
    pub a_decay_h: Tensor<B, 1>,
    pub b_bnlgr: Tensor<B, 5>,
    pub c_bnlgr: Tensor<B, 5>,
    pub d_h: Tensor<B, 1>,
    pub initial_state_bhpr: Tensor<B, 4>,
    pub init_state_hpr: Option<Tensor<B, 3>>,
}

Fields§

§x_bnlhp: Tensor<B, 5>

§Shape

[batch, nchunks, chunk_len, nheads, per_head_dim]

§dt_bnlh: Tensor<B, 4>

§Shape

[batch, nchunks, chunk_len, nheads]

§a_decay_h: Tensor<B, 1>

§Shape

[nheads]

§b_bnlgr: Tensor<B, 5>

§Shape

[batch, nchunks, chunk_len, ngroups, state_rank]

§c_bnlgr: Tensor<B, 5>

§Shape

[batch, nchunks, chunk_len, ngroups, state_rank]

§d_h: Tensor<B, 1>

§Shape

[nheads]

§initial_state_bhpr: Tensor<B, 4>

§Shape

[batch, nheads, per_head_dim, state_rank]

§init_state_hpr: Option<Tensor<B, 3>>

§Shape

[nheads, per_head_dim, state_rank]

Implementations§

impl<B: Backend> Mamba2SsdInput<B>

pub fn sanity(&self)

Auto Trait Implementations§

impl<B> Freeze for Mamba2SsdInput<B>
where <B as BackendTypes>::FloatTensorPrimitive: Freeze, <B as BackendTypes>::QuantizedTensorPrimitive: Freeze,

impl<B> RefUnwindSafe for Mamba2SsdInput<B>
where <B as BackendTypes>::FloatTensorPrimitive: RefUnwindSafe, <B as BackendTypes>::QuantizedTensorPrimitive: RefUnwindSafe,

impl<B> Send for Mamba2SsdInput<B>

impl<B> Sync for Mamba2SsdInput<B>

impl<B> Unpin for Mamba2SsdInput<B>
where <B as BackendTypes>::FloatTensorPrimitive: Unpin, <B as BackendTypes>::QuantizedTensorPrimitive: Unpin,

impl<B> UnsafeUnpin for Mamba2SsdInput<B>
where <B as BackendTypes>::FloatTensorPrimitive: UnsafeUnpin, <B as BackendTypes>::QuantizedTensorPrimitive: UnsafeUnpin,

impl<B> UnwindSafe for Mamba2SsdInput<B>
where <B as BackendTypes>::FloatTensorPrimitive: UnwindSafe, <B as BackendTypes>::QuantizedTensorPrimitive: UnwindSafe,

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.