create_linspace_latent_tokens#

torch_brain.utils.create_linspace_latent_tokens(start, end, step, num_latents_per_step)[source]#

Creates a sequence of latent tokens. Each token is defined by the latent index and the timestamps. The sequence is defined by the start and end time and the step size. The group of num_latents_per_step latents is repeated for each step.

Parameters:

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

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

  • step (float) – The step size.

  • num_latents_per_step (int) – The number of latents per step.