ml4gw.transforms.waveforms

Classes

WaveformProjector(ifos, sample_rate)
WaveformSampler([parameters])

class ml4gw.transforms.waveforms.WaveformProjector(ifos, sample_rate)

Bases: Module

Parameters:

• ifos (List[str])

• sample_rate (float)

forward(dec, psi, phi, **polarizations)

Parameters:

• dec (Float[Tensor, 'batch'])

• psi (Float[Tensor, 'batch'])

• phi (Float[Tensor, 'batch'])

• polarizations (Float[Tensor, 'batch time'])

class ml4gw.transforms.waveforms.WaveformSampler(parameters=None, **polarizations)

Bases: Module

Parameters:

• parameters (Float[Tensor, 'batch num_params'] | None)

• polarizations (Float[Tensor, 'batch time'])

forward(N)

Parameters:

N (int)