GpuDedisperserOutputs

class pirate_frb.core.GpuDedisperserOutputs

Helper class, representing dedispersion outputs for one beam batch. Return value from FrbGrouper.get_output(ichunk, ibatch)

Example usage:

with grouper.get_output(ichunk, ibatch) as outputs:
    # 'outputs' has type GpuDedisperserOutputs
    # Loop over dedispersion trees.
    for itree, tree_out in enumerate(outputs.out_max):
        # 'tree_out' has shape (beams_per_batch, coarse_ndm, coarse_ntime)

Attributes (all read-only):

• ichunk_zero_based (int) – chunk index of this output, relative to the first dedisperser output.

• ichunk_fpga_based (int) – chunk index relative to FPGA seq 0 (= ichunk_zero_based + the producer’s initial_chunk).

• ibeam (int) – beam index (NOT beam_id) of the first beam in this output.

• out_max (list) – length-ntrees list of peak-finding maximum-value arrays.

• out_argmax (list) – length-ntrees list of peak-finding argmax-token arrays.