phasespace¶
import tensorwaves.data.phasespace
Implementations of PhaseSpaceGenerator
and UniformRealNumberGenerator
.
- class TFPhaseSpaceGenerator[source]¶
Bases:
tensorwaves.interface.PhaseSpaceGenerator
Implements a phase space generator using tensorflow.
- generate(size: int, rng: UniformRealNumberGenerator) Tuple[Mapping[int, Sequence[Tuple[float, float, float, float]]], ndarray] [source]¶
Generate phase space sample.
- Returns
A
tuple
of aMomentumSample
plus a event-wise sequence of weights.
- setup(initial_state_mass: float, final_state_masses: Mapping[int, float]) None [source]¶
Hook for initialization of the
PhaseSpaceGenerator
.Called before any
generate()
calls.- Parameters
initial_state_mass – Mass of the decaying state.
final_state_masses – A mapping of final state IDs to the corresponding masses.
- class TFUniformRealNumberGenerator(seed: Optional[float] = None)[source]¶
Bases:
tensorwaves.interface.UniformRealNumberGenerator
Implements a uniform real random number generator using tensorflow.