Struct tch::nn::LayerNorm [−][src]
pub struct LayerNorm {
pub ws: Option<Tensor>,
pub bs: Option<Tensor>,
pub normalized_shape: Vec<i64>,
// some fields omitted
}
Expand description
A layer-normalization layer.
Fields
ws: Option<Tensor>
bs: Option<Tensor>
normalized_shape: Vec<i64>