code: def _get_single_X(Xs: List[Tensor]) -> Tensor:
      """Verify all X are identical, and return one.
  
      Args:
          Xs: A list of X tensors
  
      Returns: Xs[0], after verifying they are all identical.
      """
      X = Xs[0]
      for i in range(1, len(Xs)):
          assert torch.allclose(X, Xs[i])
      return X
  
imports: import dataclasses
  from typing import Any, List, Optional, Tuple
  from warnings import warn
  import torch
  from ax.core.search_space import SearchSpaceDigest
  from ax.core.types import TCandidateMetadata
  from ax.models.torch.botorch import BotorchModel
  from ax.models.torch_base import TorchGenResults, TorchModel, TorchOptConfig
  from ax.utils.common.docutils import copy_doc
  from botorch.utils.datasets import SupervisedDataset
  from torch import Tensor
label_desc: Use image node_id 1 for calling a global function with example usage: _get_single_X(Xs) and returns: X