synax.Linear.parameter_loss

synax.Linear.parameter_loss(parameters: jax.Array) jax.Array | float[source]

Parameter loss.

Parameters:

parameters – Parameters.

Returns:

Scalar.