Struct tch::nn::BatchNormConfig
source · pub struct BatchNormConfig {
pub cudnn_enabled: bool,
pub eps: f64,
pub momentum: f64,
pub affine: bool,
pub ws_init: Init,
pub bs_init: Init,
}Expand description
Batch-normalization config.
Fields§
§cudnn_enabled: bool§eps: f64§momentum: f64§affine: bool§ws_init: Init§bs_init: InitTrait Implementations§
source§impl Clone for BatchNormConfig
impl Clone for BatchNormConfig
source§fn clone(&self) -> BatchNormConfig
fn clone(&self) -> BatchNormConfig
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 more