Rate this Page
★★★★★
ReLU6#
- classtorch.nn.ReLU6(inplace=False)[source]#
Applies the ReLU6 function element-wise.
- Parameters
inplace (bool) – can optionally do the operation in-place. Default:
False
- Shape:
Input:, where means any number of dimensions.
Output:, same shape as the input.

Examples:
>>>m=nn.ReLU6()>>>input=torch.randn(2)>>>output=m(input)
On this page