Applies the rectified linear unit function element-wise.
nn_module
Calls torch::nn_relu() when trained.
Input and Output Channels
One input channel called "input" and one output channel called "output".
For an explanation see PipeOpTorch.
Super classes
mlr3pipelines::PipeOp -> PipeOpTorch -> PipeOpTorchReLU
Methods
PipeOpTorchReLU$new()
Creates a new instance of this R6 class.
Usage
PipeOpTorchReLU$new(id = "nn_relu", param_vals = list())Arguments
id(
character(1))
Identifier of the resulting object.param_vals(
list())
List of hyperparameter settings, overwriting the hyperparameter settings that would otherwise be set during construction.
Examples
# Construct the PipeOp
pipeop = nn("relu")
pipeop
#>
#> ── PipeOp <relu>: not trained ──────────────────────────────────────────────────
#> Values: list()
#>
#> ── Input channels:
#> name train predict
#> <char> <char> <char>
#> input ModelDescriptor Task
#>
#> ── Output channels:
#> name train predict
#> <char> <char> <char>
#> output ModelDescriptor Task
# The available parameters
pipeop$param_set
#> <ParamSet(1)>
#> id class lower upper nlevels default value
#> <char> <char> <num> <num> <num> <list> <list>
#> 1: inplace ParamLgl NA NA 2 FALSE [NULL]