pub struct VaeResnetRef<'a> {
pub norm1: VaeNormRef<'a>,
pub conv1: VaeConvRef<'a>,
pub norm2: VaeNormRef<'a>,
pub conv2: VaeConvRef<'a>,
pub shortcut: Option<VaeConvRef<'a>>,
}Expand description
One resnet: x + conv2(silu(norm2(conv1(silu(norm1(x)))))), with the
skip through the 1×1 shortcut when in/out channels differ.
Fields§
§norm1: VaeNormRef<'a>§conv1: VaeConvRef<'a>§norm2: VaeNormRef<'a>§conv2: VaeConvRef<'a>§shortcut: Option<VaeConvRef<'a>>Trait Implementations§
Source§impl<'a> Clone for VaeResnetRef<'a>
impl<'a> Clone for VaeResnetRef<'a>
impl<'a> Copy for VaeResnetRef<'a>
Auto Trait Implementations§
impl<'a> Freeze for VaeResnetRef<'a>
impl<'a> RefUnwindSafe for VaeResnetRef<'a>
impl<'a> Send for VaeResnetRef<'a>
impl<'a> Sync for VaeResnetRef<'a>
impl<'a> Unpin for VaeResnetRef<'a>
impl<'a> UnsafeUnpin for VaeResnetRef<'a>
impl<'a> UnwindSafe for VaeResnetRef<'a>
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