PyTorch
torch / nn / torch.nn
Threshold¶
-
class
torch.nn.
Threshold
(threshold: float, value: float, inplace: bool = False)[source]¶ Thresholds each element of the input Tensor.
Threshold is defined as:
y={x,value, if x>threshold otherwise - Parameters
threshold – The value to threshold at
value – The value to replace with
inplace – can optionally do the operation in-place. Default:
False
- Shape:
Input: (N,∗) where * means, any number of additional dimensions
Output: (N,∗) , same shape as the input
Examples:
>>> m = nn.Threshold(0.1, 20) >>> input = torch.randn(2) >>> output = m(input)
此页内容是否对您有帮助
感谢反馈!