Struct UpBlock2D 

pub struct UpBlock2D {
    pub resnets: Vec<ResnetBlock2D>,
    pub config: UpBlock2DConfig,
    /* private fields */
}

Fields

resnets: Vec<ResnetBlock2D>

config: UpBlock2DConfig

Implementations

impl UpBlock2D

pub fn new( vs: Path<'_>, in_channels: i64, prev_output_channels: i64, out_channels: i64, temb_channels: Option<i64>, config: UpBlock2DConfig, ) -> Self

pub fn forward( &self, xs: &Tensor, res_xs: &[Tensor], temb: Option<&Tensor>, upsample_size: Option<(i64, i64)>, ) -> Tensor

Trait Implementations

impl Debug for UpBlock2D

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

Formats the value using the given formatter.