pub struct VaeResnetArgs<'a> {Show 16 fields
pub groups: usize,
pub ic: usize,
pub oc: usize,
pub h: usize,
pub w: usize,
pub n1w: &'a [f32],
pub n1b: &'a [f32],
pub c1w: &'a [f32],
pub c1b: &'a [f32],
pub c1k: usize,
pub n2w: &'a [f32],
pub n2b: &'a [f32],
pub c2w: &'a [f32],
pub c2b: &'a [f32],
pub c2k: usize,
pub shortcut: Option<(&'a [f32], &'a [f32], usize)>,
}Expand description
One VAE resnet block for vae_resnet: norm/conv weights and the
channel/shape geometry. shortcut is the 1×1 projection (w, b, k)
when in/out channels differ.
Fields§
§groups: usize§ic: usize§oc: usize§h: usize§w: usize§n1w: &'a [f32]§n1b: &'a [f32]§c1w: &'a [f32]§c1b: &'a [f32]§c1k: usize§n2w: &'a [f32]§n2b: &'a [f32]§c2w: &'a [f32]§c2b: &'a [f32]§c2k: usize§shortcut: Option<(&'a [f32], &'a [f32], usize)>Auto Trait Implementations§
impl<'a> Freeze for VaeResnetArgs<'a>
impl<'a> RefUnwindSafe for VaeResnetArgs<'a>
impl<'a> Send for VaeResnetArgs<'a>
impl<'a> Sync for VaeResnetArgs<'a>
impl<'a> Unpin for VaeResnetArgs<'a>
impl<'a> UnsafeUnpin for VaeResnetArgs<'a>
impl<'a> UnwindSafe for VaeResnetArgs<'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