diffusion_rs_common::core

Trait ModuleT

source
pub trait ModuleT {
    // Required method
    fn forward_t(&self, xs: &Tensor, train: bool) -> Result<Tensor>;
}

Required Methods§

source

fn forward_t(&self, xs: &Tensor, train: bool) -> Result<Tensor>

Implementors§