Struct tch::nn::BatchNorm [−][src]
pub struct BatchNorm {
pub running_mean: Tensor,
pub running_var: Tensor,
pub ws: Tensor,
pub bs: Tensor,
pub nd: usize,
// some fields omitted
}
Expand description
A batch-normalization layer.
Fields
running_mean: Tensor
running_var: Tensor
ws: Tensor
bs: Tensor
nd: usize