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

elu

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

This is the quantized version of elu().

Parameters
  • input (Tensor) – quantized input
  • scale (float) – quantization scale of the output tensor
  • zero_point (int) – quantization zero point of the output tensor
  • alpha (float) – the alpha constant
Return type

Tensor

© 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.functional.elu.html