boax.core.models.means.linear#
- boax.core.models.means.linear(scale, bias)#
The linear mean function.
Computes y = f(x) = scale * x + bias.
Example
>>> mean = linear(jnp.array([0.2, 3.0]), 5.0) >>> ys = mean(xs)
- Parameters:
scale (
Union[Array,ndarray,bool,number]) – The d-dim scale parameter.bias (
Union[Array,ndarray,bool,number,float,int]) – The numeric bias.
- Return type:
- Returns:
A linear mean function.