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, 0.5) >>> input = torch.arange(-3, 3) >>> output = m(input)