On this page
LayerNorm
class torch.ao.nn.quantized.LayerNorm(normalized_shape, weight, bias, scale, zero_point, eps=1e-05, elementwise_affine=True, device=None, dtype=None)
[source]-
This is the quantized version of
LayerNorm
.- Additional args:
-
- scale - quantization scale of the output, type: double.
- zero_point - quantization zero point of the output, type: long.
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.ao.nn.quantized.LayerNorm.html