Enum Seq2SeqBatch 

pub enum Seq2SeqBatch<B: Backend> {
    Padded(PaddedSeq2SeqBatch<B>),
    Packed(PackedSeq2SeqBatch<B>),
}

Available on crate feature std only.

Explicit native paired device batch for padded or packed encoder-decoder models.

Variants

Padded(PaddedSeq2SeqBatch<B>)

Independently padded source and decoder input axes.

Packed(PackedSeq2SeqBatch<B>)

Independently packed paired document axes.

Implementations

impl<B: Backend> Seq2SeqBatch<B>

pub fn supervised_tokens(&self) -> usize

Actual decoder supervised labels for weighted gradient accumulation.

pub fn examples(&self) -> usize

Actual paired example count for committing the source sampler cursor.

pub fn to_device(self, device: &B::Device) -> Self

Move both source and decoder tensors to the same caller-selected device.

Trait Implementations

impl<B: Clone + Backend> Clone for Seq2SeqBatch<B>

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<B: Debug + Backend> Debug for Seq2SeqBatch<B>

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

Formats the value using the given formatter.