Sequential

class seli.net.Sequential(*modules: Callable)

Bases: Module

Call a sequence of modules in order. The first argument is updated, while the args and kwargs are passed to each module.

Parameters:

modules (Callable[..., jax.Array]) – Modules to call in order.

Methods Summary

__call__(x, *args, **kwargs)

Call self as a function.

Methods Documentation

__call__(x, *args, **kwargs)

Call self as a function.