pyhgf.typing.vectorised.Network#

class pyhgf.typing.vectorised.Network(layers, volatility_updates, max_posterior_precision, precision_clipping_value=1e-06)[source]#

Complete vectorised network state.

time_step is not stored on the network — it is passed as a per-step input to propagation_step, matching the nodalised backend’s input_data(time_steps=...) API.

Optimiser state lives in a separate optax opt-state carried alongside Network in the scan carry; it is not part of the network PyTree.

layers is a mixed tuple of Layer and LayerStack elements.

Parameters:

  • layers (tuple) – A mixed tuple of Layer and LayerStack elements, ordered from the bottom (input) layer to the top.

  • volatility_updates (str) – The volatility update scheme, e.g. "unbounded".

  • max_posterior_precision (float) – The maximum posterior precision used to clip the precision updates.

  • precision_clipping_value (float)

__init__(layers, volatility_updates, max_posterior_precision, precision_clipping_value=1e-06)#
Parameters:

  • layers (tuple)

  • volatility_updates (str)

  • max_posterior_precision (float)

  • precision_clipping_value (float)

Return type:

None

Methods

__init__(layers, volatility_updates, ...[, ...])

get_layer_sizes()

Per-element node count (one entry per Layer / LayerStack).

weights_tuple()

Per-element weights_in tuple, matched 1:1 to self.layers.

Attributes

n_layers

Number of elements (Layer or LayerStack) in the network.

n_total_slices

Total unrolled layer count, expanding every LayerStack.

params

Per-layer LayerParams tuple.

precision_clipping_value

weights

Tuple of weight matrices (legacy view).

layers

volatility_updates

max_posterior_precision