inox.nn.container

Container modules

Classes

Sequential

Creates a composition of layers.

Descriptions

class inox.nn.container.Sequential(*layers)

Creates a composition of layers.

\[y = f_n \circ \dots \circ f_2 \circ f_1(x)\]

Parameters:

layers (Module) – A sequence of layers \(f_1, f_2, \dots, f_n\).

__call__(x)

Parameters:

x (Any) – The input \(x\).

Returns:

The output \(y\).

Return type:

Any