deepfold.utils.tensor_utils.batched_gather

deepfold.utils.tensor_utils.batched_gather(data: Tensor, inds: Tensor, dim: int = 0, num_batch_dims: int = 0) → Tensor