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