pub struct UpBlock2D {
pub resnets: Vec<ResnetBlock2D>,
pub config: UpBlock2DConfig,
/* private fields */
}
Fields§
§resnets: Vec<ResnetBlock2D>
§config: UpBlock2DConfig
Implementations§
source§impl UpBlock2D
impl UpBlock2D
pub fn new( vs: VarBuilder<'_>, in_channels: usize, prev_output_channels: usize, out_channels: usize, temb_channels: Option<usize>, config: UpBlock2DConfig ) -> Result<Self>
pub fn forward( &self, xs: &Tensor, res_xs: &[Tensor], temb: Option<&Tensor>, upsample_size: Option<(usize, usize)> ) -> Result<Tensor>
Trait Implementations§
Auto Trait Implementations§
impl Freeze for UpBlock2D
impl !RefUnwindSafe for UpBlock2D
impl Send for UpBlock2D
impl Sync for UpBlock2D
impl Unpin for UpBlock2D
impl !UnwindSafe for UpBlock2D
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