Struct tch::nn::BatchNorm

pub struct BatchNorm {
    pub running_mean: Tensor,
    pub running_var: Tensor,
    pub ws: Option<Tensor>,
    pub bs: Option<Tensor>,
    pub nd: usize,
    /* private fields */
}

A batch-normalization layer.

Fields

running_mean: Tensor

running_var: Tensor

ws: Option<Tensor>

bs: Option<Tensor>

nd: usize

Trait Implementations

impl Debug for BatchNorm

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl ModuleT for BatchNorm

fn forward_t(&self, xs: &Tensor, train: bool) -> Tensor

fn batch_accuracy_for_logits( &self, xs: &Tensor, ys: &Tensor, d: Device, batch_size: i64 ) -> f64