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.