Scalar constant model#

class fortuna.model.scalar_constant.ScalarConstantModel(output_dim, initializer_fun=<function zeros>, parent=<flax.linen.module._Sentinel object>, name=None)[source]#

Bases: Module

A scalar constant model, that is \(f(\theta, x) = \theta\), with \(\theta\in\mathbb{R}\). The scalar value will be broadcasted to the output dimension.

Parameters:

  • output_dim (int) – The output model dimension.

  • initializer_fun (Optional[Initializer]) – Function to initialize the model parameters. This must be one of the available options in flax.linen.initializers.