phasespace¶

import tensorwaves.data.phasespace

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 a DataSample (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]¶

Implements a uniform real random number generator using tensorflow.

__call__(size: int, min_value: float = 0.0, max_value: float = 1.0) → ndarray[source]¶: Generate random floats in the range from [min_value,max_value).

property seed: Optional[float]¶: Get random seed. None if you want indeterministic behavior.

TensorWaves