阈值

函数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

本页目录