Struct ChineseClipTextTransformer 

pub struct ChineseClipTextTransformer {
    pub device: Device,
    /* private fields */
}

Fields

device: Device

Implementations

impl ChineseClipTextTransformer

pub fn new(var: VarBuilder<'_>, config: &ChineseClipTextConfig) -> Result<Self>

pub fn forward( &self, input_ids: &Tensor, token_type_ids: Option<&Tensor>, attention_mask: Option<&Tensor>, ) -> Result<Tensor>

Trait Implementations

impl Clone for ChineseClipTextTransformer

fn clone(&self) -> ChineseClipTextTransformer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ChineseClipTextTransformer

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

Formats the value using the given formatter.