Multiply

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

Bases: Sequential

Multiply the output of a sequence of modules by 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.