Sequential

API

class warp_nn.modules.layers.Sequential(*args)

Bases: Module

Apply callable modules (e.g. layers, activation functions, etc.) connected in a cascading sequence.

Parameters:

args – Callable modules to apply in sequence (in the order they are passed to the class constructor).

__call__(input: Any) → Any

Forward pass of the sequential container.

Parameters:

input – The input to the first module in the container.

Returns:

The output of the last module in the container.

__len__() → int

Get the number of modules in the sequential container.

Returns:

The number of registered modules.