Defines a set of torch modules composed of: (a) linear transformation; (b) batch normalization; (c) relu activation
.torch_multi_linear_batch_norm_relu(input_dim, hidden_dims)
A linear tensor block.
Input dimension of neural net.
Hidden dimensions of neural net.
Charlotte Pelletier, charlotte.pelletier@univ-ubs.fr
Gilberto Camara, gilberto.camara@inpe.br
Rolf Simoes, rolf.simoes@inpe.br
Felipe Souza, lipecaso@gmail.com