unique

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

Return the unique elements of input.

Unique elements and index where input mapping to are returned:

x = tf.constant([1, 2, 3, 2])
y, index = tf.unique(x)
print(y)  # [1, 2, 3]
print(index)  # [0, 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.