def METHOD_NAME(self, row_index: IndexType, col_index: IndexType, *batch_indices: IndexType) -> torch.Tensor:
    # Perform the __getitem__
    res = self.tensor[(*batch_indices, row_index, col_index)]
    return res