tf.unique_with_counts
| Просмотреть исходный код на GitHub |
Находит уникальные элементы в 1-мерном тензоре.
tf.unique_with_counts(
x,
out_idx=tf.dtypes.int32,
name=None
)
Эта операция возвращает тензор y , содержащий все уникальные элементы x , отсортированные в том же порядке, в котором они встречаются в x. Эта операция также возвращает тензор idx того же размера, что и x, который содержит индекс каждого значения x в выходном уникальном тензоре y. Наконец, она возвращает третий тензор count, содержащий количество каждого элемента 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, 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]
| Аргументы | |
|---|---|
x | A Tensor. 1-D. |
out_idx | Необязательный tf.DType из: tf.int32, tf.int64. По умолчанию tf.int32. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Кортеж объектов Tensor (y, idx, count). | |
y | A Tensor. Имеет тот же тип, что и x. |
idx | A Tensor типа out_idx. |
count | A Tensor типа out_idx. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/unique_with_counts