Function kn_graph::cpu::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.