Struct candle_transformers::models::wuerstchen::prior::WPrior
source · pub struct WPrior { /* private fields */ }
Implementations§
source§impl WPrior
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§
Auto Trait Implementations§
impl Freeze for WPrior
impl !RefUnwindSafe for WPrior
impl Send for WPrior
impl Sync for WPrior
impl Unpin for WPrior
impl !UnwindSafe for WPrior
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more