pub struct EuclideanCodebook<T: Tensor> {
pub _initialized: Parameter<T>,
pub cluster_usage: Parameter<T>,
pub embedding_sum: Parameter<T>,
}Expand description
Euclidean codebook backed by EMA cluster statistics.
Fields§
§_initialized: Parameter<T>Checkpoint initialization flag.
cluster_usage: Parameter<T>EMA cluster usage.
embedding_sum: Parameter<T>EMA embedding sum.
Trait Implementations§
Auto Trait Implementations§
impl<T> Freeze for EuclideanCodebook<T>
impl<T> RefUnwindSafe for EuclideanCodebook<T>where
Parameter<T>: RefUnwindSafe,
impl<T> Send for EuclideanCodebook<T>
impl<T> Sync for EuclideanCodebook<T>
impl<T> Unpin for EuclideanCodebook<T>
impl<T> UnsafeUnpin for EuclideanCodebook<T>where
Parameter<T>: UnsafeUnpin,
impl<T> UnwindSafe for EuclideanCodebook<T>where
Parameter<T>: UnwindSafe,
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