TensorFlow 計(jì)算整數(shù)數(shù)組值出現(xiàn)的次數(shù)

tf.bincount

bincount ( 
    arr , 
    weights = None, 
    minlength = None, 
    maxlength = None, 
    dtype = tf.int32
 )

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

計(jì)算整數(shù)數(shù)組中每個(gè)值的出現(xiàn)次數(shù).

如果 minlength 與 maxlength 沒(méi)有給出,返回與長(zhǎng)度的矢量 tf.reduce_max(arr) + 1,如果發(fā)射是非空的,則長(zhǎng)度為0.如果權(quán)重是非零, 則輸出的索引 i 存儲(chǔ)每個(gè)索引中的權(quán)重值之和, 其中的相應(yīng)值是 i.

ARGS：

• arr：非負(fù)值的 int32 張量.
• weights：如果非 None,則與 arr 的形狀必須相同.對(duì)于每個(gè) arr 值,bin 將以相應(yīng)的權(quán)重遞增而不是1.
• minlength：如果給定,確保輸出至少具有長(zhǎng)度 minlength ,必要時(shí)在末尾填充零.
• maxlength：如果給定,在 arr 中跳過(guò)等于或大于的值 maxlength,確保輸出的長(zhǎng)度最多為 maxlength.
• dtype：如果權(quán)重為 None,則確定輸出 bins 的類型.

返回：

返回與權(quán)重或給定的 dtype 相同的 dtype 的向量.bin 值.