Bias

Bias#

class seli.net.Bias[source]#

Bases: Module

Add a learnable bias to the last axis of the input.

Attributes Summary

dim

Return the dimension of the bias.

Methods Summary

__call__(x)

Call self as a function.

Attributes Documentation

dim#

Return the dimension of the bias. If the bias has not been initialized yet, return None.

Methods Documentation

__call__(x: Float[Array, '*batch dim']) Float[Array, '*batch dim'][source]#

Call self as a function.