threshold

class torch.ao.nn.quantized.functional.threshold(input, threshold, value)

逐元素应用阈值函数的量化版本

$$x = \begin{cases} x & \text{if~} x > \text{threshold} \\ \text{value} & \text{otherwise} \end{cases}$$

有关更多详细信息，请参阅 Threshold。

返回类型

Tensor