TensorFlow函數(shù)：tf.unique

tf.unique函數(shù)

tf.unique(
    x,
    out_idx=tf.int32,
    name=None
)

定義在：tensorflow/python/ops/array_ops.py

請參閱指南：數(shù)學(xué)函數(shù)>序列比較和索引

在一維張量中找到唯一的元素.

該操作返回一個張量 y,該張量包含所有發(fā)生在 x 中的所有唯一的元素 x,它們按照相同的順序排序.此操作還會返回一個與 x 具有相同大小的張量 idx,包含唯一的輸出 y 中 x 的每個值的索引.也就是說：

y[idx[i]] = x[i] for i in [0, 1,...,rank(x) - 1]

例如：

# tensor 'x' is [1, 1, 2, 4, 4, 4, 7, 8, 8]
y, idx = unique(x)
y ==> [1, 2, 4, 7, 8]
idx ==> [0, 0, 1, 2, 2, 2, 3, 4, 4]

函數(shù)參數(shù)：

• x：一個 Tensor,是1維的.
• out_idx：可選 tf.DType 來自：tf.int32, tf.int64,默認為 tf.int32.
• name：操作的名稱(可選).

函數(shù)返回值：

Tensor對象(y, idx)的元型態(tài)組.

• y：一個 Tensor,與 x 類型相同.
• idx：一個 out_idx 類型的 Tensor.