Shortcuts

LeakyReLU

class torch.ao.nn.quantized.LeakyReLU(scale, zero_point, negative_slope=0.01, inplace=False, device=None, dtype=None)[源代码]

这是 LeakyReLU 的量化等效形式。

Parameters
  • scale (float) – 输出张量的量化比例

  • zero_point (int) – 输出张量的量化零点

  • negative_slope (float) – 控制负斜率的角度。默认值:1e-2

优云智算