Struct dfdx::nn::builders::TransformerEncoder

pub struct TransformerEncoder<const MODEL_DIM: usize, const NUM_HEADS: usize, const FF_DIM: usize, const NUM_LAYERS: usize>;

Trait Implementations

impl<const M: usize, const H: usize, const F: usize, const L: usize, E: Dtype, D: Device<E>> BuildOnDevice<D, E> for TransformerEncoder<M, H, F, L>where TransformerEncoder<M, H, F, L, E, D>: BuildModule<D, E>,

type Built = Repeated<TransformerEncoderBlock<M, H, F, E, D>, L>

fn try_build_on_device(device: &D) -> Result<Self::Built, D::Err>

fn build_on_device(device: &D) -> Self::Built

impl<const MODEL_DIM: usize, const NUM_HEADS: usize, const FF_DIM: usize, const NUM_LAYERS: usize> Debug for TransformerEncoder<MODEL_DIM, NUM_HEADS, FF_DIM, NUM_LAYERS>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.