get_subset_tensor#

deepinv.physics.functional.tomography_subsets.get_subset_tensor(tensor, num_subsets)[source]#

Return indices that interleave a tensor into equal subsets.

Parameters:

  • tensor (torch.Tensor) – tensor containing angles or geometry vectors to split along its first dimension.

  • num_subsets (int) – number of subsets.

Returns:

list of index tensors.

Return type:

list[Tensor]