Skip to contents

Gelu

Input and Output Channels

One input channel called "input" and one output channel called "output". For an explanation see PipeOpTorch.

State

The state is the value calculated by the public method $shapes_out().

Credit

Part of this documentation have been copied or adapted from the documentation of torch.

Parameters

  • approximate :: character(1)
    Whether to use an approximation algorithm. Default is "none".

Internals

Calls torch::nn_gelu() when trained.

Super classes

mlr3pipelines::PipeOp -> mlr3torch::PipeOpTorch -> PipeOpTorchGELU

Methods

Inherited methods


Method new()

Creates a new instance of this R6 class.

Usage

PipeOpTorchGELU$new(id = "nn_gelu", 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.


Method clone()

The objects of this class are cloneable with this method.

Usage

PipeOpTorchGELU$clone(deep = FALSE)

Arguments

deep

Whether to make a deep clone.

Examples

# Construct the PipeOp
pipeop = po("nn_gelu")
pipeop
#> PipeOp: <nn_gelu> (not trained)
#> values: <list()>
#> Input channels <name [train type, predict type]>:
#>   input [ModelDescriptor,Task]
#> Output channels <name [train type, predict type]>:
#>   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: approximate ParamFct    NA    NA       2    none [NULL]