nn_nonlinear: Shortcut to create linear layer with nonlinear activation...

In krzjoa/torchts: Time series Models with torch

Shortcut to create linear layer with nonlinear activation function

Description

Shortcut to create linear layer with nonlinear activation function

Usage

nn_nonlinear(in_features, out_features, bias = TRUE, activation = nn_relu())

Arguments

in_features	(integer) size of each input sample
out_features	(integer) size of each output sample
bias	(logical) If set to FALSE, the layer will not learn an additive bias. Default: TRUE
activation	(nn_module) A nonlinear activation function (default: torch::nn_relu())

Examples

net <- nn_nonlinear(10, 1)
x   <- torch_tensor(matrix(1, nrow = 2, ncol = 10))
net(x)

krzjoa/torchts documentation built on June 24, 2022, 5:30 a.m.