Struct ClipEncoder

pub struct ClipEncoder { /* private fields */ }

Implementations

impl ClipEncoder

pub fn new(vs: VarBuilder<'_>, c: &EncoderConfig) -> Result<Self>

pub fn forward( &self, xs: &Tensor, causal_attention_mask: Option<&Tensor>, ) -> Result<Tensor>

pub fn output_hidden_states( &self, xs: &Tensor, causal_attention_mask: Option<&Tensor>, ) -> Result<Vec<Tensor>>

Trait Implementations

impl Clone for ClipEncoder

fn clone(&self) -> ClipEncoder

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ClipEncoder

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

Formats the value using the given formatter.