Function tch::nn::layer_norm
source · pub fn layer_norm<'a, T: Borrow<Path<'a>>>(
vs: T,
normalized_shape: Vec<i64>,
config: LayerNormConfig
) -> LayerNorm
pub fn layer_norm<'a, T: Borrow<Path<'a>>>(
vs: T,
normalized_shape: Vec<i64>,
config: LayerNormConfig
) -> LayerNorm