Threshold¶
- class torch.nn.Threshold(threshold, value, inplace=False)[source]¶
Thresholds each element of the input Tensor.
Threshold is defined as:
- Parameters
- Shape:
Input: , where means any number of dimensions.
Output: , same shape as the input.
Examples:
>>> m = nn.Threshold(0.1, 20) >>> input = torch.randn(2) >>> output = m(input)