阈值¶ class torch.nn.Threshold(threshold, value, inplace=False)[source]¶ 对输入张量的每个元素进行阈值处理。 阈值定义为 y={x, if x>thresholdvalue, otherwise y = \begin{cases} x, &\text{ if } x > \text{threshold} \\ \text{value}, &\text{ otherwise } \end{cases} y={x,value, if x>threshold otherwise 参数 threshold (float) – 阈值 value (float) – 替换的值 inplace (bool) – 可以选择就地执行操作。默认值:False 形状 输入:(∗)(*)(∗),其中 ∗*∗ 表示任意数量的维度。 输出:(∗)(*)(∗),与输入相同形状。 示例 >>> m = nn.Threshold(0.1, 20) >>> input = torch.randn(2) >>> output = m(input)