阈值 ¶

class torch.nn.Threshold(阈值, 值, inplace=False)[source][source] ¶

阈值每个输入张量的元素。

阈值定义为：

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

参数:

• 阈值（浮点数）- 阈值值

• 值（浮点数）- 替换的值

• inplace（布尔值）- 可选地进行原地操作。默认： False

形状：

• 输入： $(*)$ ，其中 $*$ 表示任意数量的维度。

• 输出： $(*)$ ，与输入具有相同的形状。

示例：

>>> m = nn.Threshold(0.1, 20)
>>> input = torch.randn(2)
>>> output = m(input)