def _get_tensor_data()

in tensorflow_gnn/graph/graph_tensor_pprint.py [0:0]


def _get_tensor_data(listorarray: Any) -> Any:
  """Convert a tensor to plain-old data types."""
  if isinstance(listorarray, (int, float, str)):
    return listorarray
  elif isinstance(listorarray, bytes):
    try:
      return listorarray.decode('utf8')
    except UnicodeDecodeError:
      # Note: This may be useful for images and other non-text binary data.
      return listorarray
  elif isinstance(listorarray, list):
    return [_get_tensor_data(elem) for elem in listorarray]
  elif isinstance(listorarray, numpy.ndarray):
    return [_get_tensor_data(elem) for elem in listorarray.tolist()]
  else:
    raise TypeError(f'Unsupported type: {listorarray}')