Function torch::nn::functional::layer_norm

• Defined in File normalization.h

Function Documentation

inline Tensor torch::nn::functional::layer_norm(const Tensor &input, const LayerNormFuncOptions &options)

See https://pytorch.org/docs/main/nn.functional.html#torch.nn.functional.layer_norm about the exact behavior of this functional.

See the documentation for torch::nn::functional::LayerNormFuncOptions class to learn what optional arguments are supported for this functional.

Example:

namespace F = torch::nn::functional;
F::layer_norm(input, F::LayerNormFuncOptions({2, 2}).eps(2e-5));