torch (version 0.0.2)

nn_sequential: A sequential container

Description

A sequential container. Modules will be added to it in the order they are passed in the constructor. See examples.

Usage

nn_sequential(..., name = NULL)

Arguments

...

sequence of modules to be added

name

optional name for the generated module.

Examples

# NOT RUN {
if (torch_is_installed()) {

model <- nn_sequential(
  nn_conv2d(1, 20, 5),
  nn_relu(),
  nn_conv2d(20, 64, 5),
  nn_relu()
)
input <- torch_randn(32, 1, 28, 28)
output <- model(input)

}
# }