Add

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

Bases: Sequential

Add the output of a sequence of modules to the input. 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.