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

ELU

class torch.ao.nn.quantized.ELU(scale, zero_point, alpha=1.0) [source]

This is the quantized equivalent of ELU.

Parameters
  • scale – quantization scale of the output tensor
  • zero_point – quantization zero point of the output tensor
  • alpha (float) – the alpha constant

© 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.ELU.html