TensorFlow函數(shù)：tf.unique_with_counts

tf.unique_with_counts函數(shù)

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

參見指南：張量變換>分割和連接

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

該操作返回一個張量 y,該張量包含出現(xiàn)在 x 中的以相同順序排序的 x 的所有的唯一元素.此操作還會返回一個與 x 具有相同大小的張量 idx,包含唯一的輸出 y 中 x 的每個值的索引.最后,它返回一個包含第三個張量 count,其中包含 x 中 y 的每個元素的計數(shù),即：

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, count = unique_with_counts(x)
y ==> [1, 2, 4, 7, 8]
idx ==> [0, 0, 1, 2, 2, 2, 3, 4, 4]
count ==> [2, 1, 3, 1, 2]

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

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

函數(shù)返回值：

Tensor 對象的元組(y,idx,count).

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