(class_ids)
return tf.unique(class_ids)[0]
Computes the unique class IDs of the episode containing `class_ids`.

  Args:
    class_ids: A 1D tensor representing class IDs, one per example in an
      episode.

  Returns:
    A 1D tensor of the unique class IDs whose size is equal to the way of an
    episode.
Computes the unique class IDs of the episode containing `class_ids`.
Computes the unique class IDs of the episode containing class_ids .
def compute_unique_class_ids(class_ids):
  """Computes the unique class IDs of the episode containing `class_ids`.

  Args:
    class_ids: A 1D tensor representing class IDs, one per example in an
      episode.

  Returns:
    A 1D tensor of the unique class IDs whose size is equal to the way of an
    episode.
  """
  return tf.unique(class_ids)[0]
def compute_unique_class_ids ( class_ids ) : return tf . unique ( class_ids ) [ 0 ]