JointBlock

candle_transformers::models::mmdit::blocks

Trait JointBlock 

Source 
pub trait JointBlock {
    // Required method
    fn forward(
        &self,
        context: &Tensor,
        x: &Tensor,
        c: &Tensor,
    ) -> Result<(Tensor, Tensor)>;
}

Required Methods§

Source

fn forward( &self, context: &Tensor, x: &Tensor, c: &Tensor, ) -> Result<(Tensor, Tensor)>

Implementors§