tsicl.utils.utils.make_grid

tsicl.utils.utils.make_grid(grid_length: int, num_samples: int, end_point: float = 1.0, start_point: float = 0.0) → torch.Tensor

Create a grid of time coords in (0, 1)

Parameters:

• grid_length (int) – number of timesteps

• num_samples (int) – number of samples

• end_point (float) – end point of the grid

• start_point (float) – start point of the grid

Returns:

Grid as torch.Tensor of shape (num_samples, grid_length, 1).