Learn R Programming

sits (version 1.1.0)

.torch_multi_linear_batch_norm_relu: Torch module for linear MLP

Description

Defines a set of torch modules composed of: (a) linear transformation; (b) batch normalization; (c) relu activation

Usage

.torch_multi_linear_batch_norm_relu(input_dim, hidden_dims)

Value

A linear tensor block.

Arguments

input_dim

Input dimension of neural net.

hidden_dims

Hidden dimensions of neural net.

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