kn_graph::cpu

Function layernorm

Source 
pub fn layernorm<S, D>(
    array: ArrayBase<S, D>,
    axis: Axis,
    eps: f32,
) -> Array<f32, D>
where
    D: RemoveAxis,
    S: RawData + Data + RawData<Elem = f32>,
Expand description

Layernorm along the given axis of the tensor.