Activation Function Arguments Helper

This is superseded in v0.3.0 in favour of indexed syntax, e.g. <act_fn[param = 0]> type. Type-safe helper to specify parameters for activation functions. All parameters must be named and match the formal arguments of the corresponding {torch} activation function.

Usage

args(...)

Arguments

...

Named arguments for the activation function.

Value

A list with class "activation_args" containing the parameters.

Examples

act_funs(softmax = args(dim = 2L))
#> <activation_spec[1]>
#> softmax 
#>       2