hard_sigmoid¶

Piecewise-linear approximation of sigmoid:

\[\text{HardSigmoid}(x) = \frac{\min(\max(x + 3,\ 0),\ 6)}{6}\]

ATen source: aten.hardsigmoid

Inputs¶

Name	Description
`input`	Tensor of any rank

Name	Type	Description
`version`	`int`	Composite op version

Name	Shape	Description
`output`	same as `input`	Same shape and dtype as `input`

fp16, fp32, bf16.

import torch

input = torch.rand(1, 3, 64, 64)
output = torch.nn.functional.hardsigmoid(input)