pub struct AttentionBlockConfig {
pub num_head_channels: Option<usize>,
pub num_groups: usize,
pub rescale_output_factor: f64,
pub eps: f64,
}
Expand description
Configuration for an attention block.
Fields§
§num_head_channels: Option<usize>
§num_groups: usize
§rescale_output_factor: f64
§eps: f64
Trait Implementations§
source§impl Clone for AttentionBlockConfig
impl Clone for AttentionBlockConfig
source§fn clone(&self) -> AttentionBlockConfig
fn clone(&self) -> AttentionBlockConfig
Returns a copy of the value. Read more
1.0.0 · source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source
. Read moresource§impl Debug for AttentionBlockConfig
impl Debug for AttentionBlockConfig
source§impl Default for AttentionBlockConfig
impl Default for AttentionBlockConfig
impl Copy for AttentionBlockConfig
Auto Trait Implementations§
impl Freeze for AttentionBlockConfig
impl RefUnwindSafe for AttentionBlockConfig
impl Send for AttentionBlockConfig
impl Sync for AttentionBlockConfig
impl Unpin for AttentionBlockConfig
impl UnwindSafe for AttentionBlockConfig
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