Struct dfdx::nn::builders::MultiHeadAttention
source · pub struct MultiHeadAttention<const EMBED_DIM: usize, const NUM_HEADS: usize, const K_DIM: usize = EMBED_DIM, const V_DIM: usize = EMBED_DIM>;
Implementations§
source§impl<const M: usize, const H: usize, const K: usize, const V: usize> MultiHeadAttention<M, H, K, V>
impl<const M: usize, const H: usize, const K: usize, const V: usize> MultiHeadAttention<M, H, K, V>
pub const TYPE_CHECK: () = _
Trait Implementations§
source§impl<const M: usize, const H: usize, const K: usize, const V: usize, E: Dtype, D: Device<E>> BuildOnDevice<D, E> for MultiHeadAttention<M, H, K, V>where
MultiHeadAttention<M, H, K, V, E, D>: BuildModule<D, E>,
impl<const M: usize, const H: usize, const K: usize, const V: usize, E: Dtype, D: Device<E>> BuildOnDevice<D, E> for MultiHeadAttention<M, H, K, V>where MultiHeadAttention<M, H, K, V, E, D>: BuildModule<D, E>,
type Built = MultiHeadAttention<M, H, K, V, E, D>
fn try_build_on_device(device: &D) -> Result<Self::Built, D::Err>
fn build_on_device(device: &D) -> Self::Built
source§impl<const EMBED_DIM: usize, const NUM_HEADS: usize, const K_DIM: usize, const V_DIM: usize> Clone for MultiHeadAttention<EMBED_DIM, NUM_HEADS, K_DIM, V_DIM>
impl<const EMBED_DIM: usize, const NUM_HEADS: usize, const K_DIM: usize, const V_DIM: usize> Clone for MultiHeadAttention<EMBED_DIM, NUM_HEADS, K_DIM, V_DIM>
source§fn clone(&self) -> MultiHeadAttention<EMBED_DIM, NUM_HEADS, K_DIM, V_DIM>
fn clone(&self) -> MultiHeadAttention<EMBED_DIM, NUM_HEADS, K_DIM, V_DIM>
Returns a copy of the value. Read more
1.0.0 · source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source
. Read moreAuto Trait Implementations§
impl<const EMBED_DIM: usize, const NUM_HEADS: usize, const K_DIM: usize, const V_DIM: usize> RefUnwindSafe for MultiHeadAttention<EMBED_DIM, NUM_HEADS, K_DIM, V_DIM>
impl<const EMBED_DIM: usize, const NUM_HEADS: usize, const K_DIM: usize, const V_DIM: usize> Send for MultiHeadAttention<EMBED_DIM, NUM_HEADS, K_DIM, V_DIM>
impl<const EMBED_DIM: usize, const NUM_HEADS: usize, const K_DIM: usize, const V_DIM: usize> Sync for MultiHeadAttention<EMBED_DIM, NUM_HEADS, K_DIM, V_DIM>
impl<const EMBED_DIM: usize, const NUM_HEADS: usize, const K_DIM: usize, const V_DIM: usize> Unpin for MultiHeadAttention<EMBED_DIM, NUM_HEADS, K_DIM, V_DIM>
impl<const EMBED_DIM: usize, const NUM_HEADS: usize, const K_DIM: usize, const V_DIM: usize> UnwindSafe for MultiHeadAttention<EMBED_DIM, NUM_HEADS, K_DIM, V_DIM>
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more