aeppl.scan.MeasurableScan.get_oinp_iinp_iout_oout_mappings#

MeasurableScan.get_oinp_iinp_iout_oout_mappings()#

Compute and return dictionary mappings between the inputs and outputs of the inner function and the inputs and outputs of the Scan node in the outer graph.

The return value is a dictionary in which the keys are the names of the individual mappings and the values are the mapping dictionaries themselves. In dictionaries representing mappings to outer variables, the values are individual integer indices. In dictionaries representing mappings to inner variables, the values are sequences of indices because multiple inner variables can be associated with the same state.