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

Hardswish

class torch.ao.nn.quantized.Hardswish(scale, zero_point, device=None, dtype=None) [source]

This is the quantized version of Hardswish.

Parameters
  • scale – quantization scale of the output tensor
  • zero_point – quantization zero point of the output 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.Hardswish.html