pytorch / 2 / generated / torch.ao.nn.quantized.layernorm.html

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