Learn R Programming

sits (version 1.1.0)

.torch_conv1D_batch_norm_relu_dropout: Torch module for Conv1D + Batch Norm + Relu + Dropout

Description

Defines a torch conv1d module composed of: (a) 1d convolution; (b) batch normalization; (c) relu activation; (d) dropout;

Usage

.torch_conv1D_batch_norm_relu_dropout(
  input_dim,
  output_dim,
  kernel_size,
  padding = 0,
  dropout_rate
)

Value

A conv1D tensor block.

Arguments

input_dim

Input dimension of neural net.

output_dim

Output dimension of neural net.

kernel_size

Size of 1D convolutional kernel.

padding

Padding added to both sides of the input.

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