create_start_end_unit_tokens#

torch_brain.utils.create_start_end_unit_tokens(unit_ids, start, end)[source]#

Creates for each unit a start and end token. Each token is defined by the unit index, the token type index and the timestamps.

Parameters:

  • unit_ids (np.ndarray) – List of unit identifiers.

  • start (float) – The start time of the sequence.

  • end (float) – The end time of the sequence.