ReLU6

class torch.nn.ReLU6(inplace=False)

逐元素地應用 ReLU6 函數。

$$\text{ReLU6}(x) = \min(\max(0,x), 6)$$

參數

inplace (bool) – 選擇性地進行原地 (in-place) 操作。預設值：False

形狀

• 輸入：$(*)$，其中 $*$ 代表任意數量的維度。

• 輸出：$(*)$，與輸入相同的形狀。

範例

>>> m = nn.ReLU6()
>>> input = torch.randn(2)
>>> output = m(input)