Rate this Page

Function at::layer_norm#

Function Documentation#

inline at::Tensor at::layer_norm(const at::Tensor &input, at::IntArrayRef normalized_shape, const ::std::optional<at::Tensor> &weight = {}, const ::std::optional<at::Tensor> &bias = {}, double eps = 1e-05, bool cudnn_enable = true)#