nn_sequential: A sequential container

In torch: Tensors and Neural Networks with 'GPU' Acceleration

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(...)

Arguments

...	sequence of modules to be added

Examples

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)
}

torch documentation built on Jan. 24, 2023, 1:05 a.m.