Learn R Programming

sits (version 1.1.0)

.torch_linear_batch_norm_relu_dropout: Torch module for linear MLP

Description

Defines a torch module composed of; (a) linear transformation; (b) batch normalization; (c) relu activation; (d) dropout

Usage

.torch_linear_batch_norm_relu_dropout(input_dim, output_dim, dropout_rate)

Value

A linear tensor block.

Arguments

input_dim

Input dimension of neural net.

output_dim

Output dimension of neural net.

dropout_rate

Dropout rate for linear module.

Author

Charlotte Pelletier, charlotte.pelletier@univ-ubs.fr

Gilberto Camara, gilberto.camara@inpe.br

Rolf Simoes, rolf.simoes@inpe.br

Felipe Souza, lipecaso@gmail.com