Struct dfdx::nn::builders::Transformer
source · pub struct Transformer<const MODEL_DIM: usize, const NUM_HEADS: usize, const NUM_ENCODER_LAYERS: usize, const NUM_DECODER_LAYERS: usize, const FF_DIM: usize>;
Trait Implementations§
source§impl<const M: usize, const H: usize, const A: usize, const B: usize, const F: usize, E, D> BuildOnDevice<D, E> for Transformer<M, H, A, B, F>where
E: Dtype,
D: Device<E>,
Transformer<M, H, A, B, F, E, D>: BuildModule<D, E>,
impl<const M: usize, const H: usize, const A: usize, const B: usize, const F: usize, E, D> BuildOnDevice<D, E> for Transformer<M, H, A, B, F>where E: Dtype, D: Device<E>, Transformer<M, H, A, B, F, E, D>: BuildModule<D, E>,
type Built = Transformer<M, H, A, B, F, 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 MODEL_DIM: usize, const NUM_HEADS: usize, const NUM_ENCODER_LAYERS: usize, const NUM_DECODER_LAYERS: usize, const FF_DIM: usize> Clone for Transformer<MODEL_DIM, NUM_HEADS, NUM_ENCODER_LAYERS, NUM_DECODER_LAYERS, FF_DIM>
impl<const MODEL_DIM: usize, const NUM_HEADS: usize, const NUM_ENCODER_LAYERS: usize, const NUM_DECODER_LAYERS: usize, const FF_DIM: usize> Clone for Transformer<MODEL_DIM, NUM_HEADS, NUM_ENCODER_LAYERS, NUM_DECODER_LAYERS, FF_DIM>
source§fn clone(
&self
) -> Transformer<MODEL_DIM, NUM_HEADS, NUM_ENCODER_LAYERS, NUM_DECODER_LAYERS, FF_DIM>
fn clone( &self ) -> Transformer<MODEL_DIM, NUM_HEADS, NUM_ENCODER_LAYERS, NUM_DECODER_LAYERS, FF_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 more