Struct WPrior 

pub struct WPrior { /* private fields */ }

Implementations

impl WPrior

pub fn new( c_in: usize, c: usize, c_cond: usize, c_r: usize, depth: usize, nhead: usize, use_flash_attn: bool, vb: VarBuilder<'_>, ) -> Result<Self>

pub fn gen_r_embedding(&self, r: &Tensor) -> Result<Tensor>

pub fn forward(&self, xs: &Tensor, r: &Tensor, c: &Tensor) -> Result<Tensor>

Trait Implementations

impl Debug for WPrior

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

Formats the value using the given formatter.