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[Union[int, str], ndarray], ndarray] [source]¶
Generate phase space sample.
- Returns
A
tuple
of aDataSample
(four-momenta) with an event-wise sequence of weights. The four-momenta are arrays of shape \(n \times 4\).
- 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.