Affine

Affine#

class seli.net.Affine(dim: int)[source]#

Bases: Module

Apply a learnable linear transformation followed by a learnable bias.

Parameters:

dim (int) – The output dimension of the linear transformation. The input dimension is inferred from the last axis of the first input.

Attributes Summary

dim_in

Methods Summary

__call__(x)

Call self as a function.

Attributes Documentation

dim_in#

Methods Documentation

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

Call self as a function.

Contents