tf.sparse.bincount

Подсчёт числа появлений целого значения в тензоре.

tf.sparse.bincount(
    values,
    weights=None,
    axis=0,
    minlength=None,
    maxlength=None,
    binary_output=False,
    name=None
)

Этот оператор принимает N-мерный Tensor, RaggedTensor, или SparseTensor, и возвращает N-мерный int64 SparseTensor, где элемент [i0...i[axis], j] содержит число появлений значения j в срезе [i0...i[axis], :] тензора входных данных. В настоящее время поддерживаются только N=0 и N=-1.

Примеры:

Подсчёт ячеек для каждого элемента в отдельных группах

В этом примере входные данные (которыми могут быть Tensor, RaggedTensor или SparseTensor) преобразуются в SparseTensor, где значение (i, j) равно количеству появлений значения j в группе i.

data = np.array([[10, 20, 30, 20], [11, 101, 11, 10001]], dtype=np.int64)
output = tf.sparse.bincount(data, axis=-1)
print(output)
SparseTensor(indices=tf.Tensor(
[[    0    10]
 [    0    20]
 [    0    30]
 [    1    11]
 [    1   101]
 [    1 10001]], shape=(6, 2), dtype=int64),
 values=tf.Tensor([1 2 1 2 1 1], shape=(6,), dtype=int64),
 dense_shape=tf.Tensor([    2 10002], shape=(2,), dtype=int64))

Подсчёт ячеек с заданной формой выходного тензора

В этом примере входные данные (которыми могут быть Tensor, RaggedTensor или SparseTensor) преобразуются в SparseTensor, где значение (i, j) равно количеству появлений значения j в группе i. Однако все значения j, превышающие 'maxlength', игнорируются. Форма dense_shape выходного SparseTensor установлена в 'minlength'. Обратите внимание, что, хотя входные данные идентичны предыдущему примеру, значение '10001' в элементе группы 2 отброшено, а форма dense установлена как [2, 500], а не [2,10002] или [2, 102].

minlength = maxlength = 500
data = np.array([[10, 20, 30, 20], [11, 101, 11, 10001]], dtype=np.int64)
output = tf.sparse.bincount(
   data, axis=-1, minlength=minlength, maxlength=maxlength)
print(output)
SparseTensor(indices=tf.Tensor(
[[  0  10]
 [  0  20]
 [  0  30]
 [  1  11]
 [  1 101]], shape=(5, 2), dtype=int64),
 values=tf.Tensor([1 2 1 2 1], shape=(5,), dtype=int64),
 dense_shape=tf.Tensor([  2 500], shape=(2,), dtype=int64))

Бинарный подсчёт ячеек

В этом примере входные данные (которые могут быть тензором, RaggedTensor или SparseTensor) преобразуются в SparseTensor, где (i, j) равно 1, если значение j появляется в группе i хотя бы один раз, и 0 в противном случае. Обратите внимание, что, хотя некоторые значения (например, 20 в группе 1 и 11 в группе 2) встречаются более одного раза, значения в тензоре 'values' составляют все 1.

data = np.array([[10, 20, 30, 20], [11, 101, 11, 10001]], dtype=np.int64)
output = tf.sparse.bincount(data, binary_output=True, axis=-1)
print(output)
SparseTensor(indices=tf.Tensor(
[[    0    10]
 [    0    20]
 [    0    30]
 [    1    11]
 [    1   101]
 [    1 10001]], shape=(6, 2), dtype=int64),
 values=tf.Tensor([1 1 1 1 1 1], shape=(6,), dtype=int64),
 dense_shape=tf.Tensor([    2 10002], shape=(2,), dtype=int64))

Подсчёт ячеек с весами

В этом примере используются два входных тензора — тензор значений и тензор весов. Эти тензоры должны иметь одинаковую форму и те же разделы строк или индексы в случае RaggedTensors или SparseTensors. При выполнении подсчёта с весами оператор выдаст SparseTensor, где значение (i, j) равно сумме значений в тензоре весов в группе i, в местах, где тензор значений имеет значение j. В этом случае тип выходного тензора совпадает с типом тензора весов.

data = np.array([[10, 20, 30, 20], [11, 101, 11, 10001]], dtype=np.int64)
weights = [[2, 0.25, 15, 0.5], [2, 17, 3, 0.9]]
output = tf.sparse.bincount(data, weights=weights, axis=-1)
print(output)
SparseTensor(indices=tf.Tensor(
[[    0    10]
 [    0    20]
 [    0    30]
 [    1    11]
 [    1   101]
 [    1 10001]], shape=(6, 2), dtype=int64),
 values=tf.Tensor([2. 0.75 15. 5. 17. 0.9], shape=(6,), dtype=float32),
 dense_shape=tf.Tensor([    2 10002], shape=(2,), dtype=int64))