unique_with_counts

dragon.vm.tensorflow.unique_with_counts(
  x,
  name=None,
  **kwargs
)[source]

Return the unique elements of input with counts.

Unique elements, remapping index and counts are returned:

x = tf.constant([1, 2, 3, 2])
y, index, counts = tf.unique_with_counts(x)
print(y)  # [1, 2, 3]
print(index)  # [0, 1, 2, 1]
print(counts)  # [1, 2, 1]
Parameters:
  • x (dragon.Tensor) – The input tensor.
  • name (str, optional) – The operation name.
Returns:

  • dragon.Tensor – The output tensor.
  • dragon.Tensor – The inverse index tensor.
  • dragon.Tensor – The counts tensor.