Python tf.math.count_nonzero用法及代碼示例

計算張量維度上非零元素的數量。

用法

tf.math.count_nonzero(
    input, axis=None, keepdims=None, dtype=tf.dtypes.int64, name=None
)

參數

input 要減少的張量。應該是數字類型，bool 或 string。
axis 要減小的尺寸。如果None(默認)，減少所有維度。必須在 [-rank(input), rank(input)) 範圍內。
keepdims 如果為真，則保留長度為 1 的縮減維度。
dtype 輸出數據類型；默認為 tf.int64 。
name 操作的名稱(可選)。

減少的張量(非零值的數量)。

沿 axis 中給定的尺寸減少 input 。除非 keepdims 為真，否則對於 axis 中的每個條目，張量的秩都會減少 1。如果 keepdims 為真，則保留縮減後的維度，長度為 1。

如果axis 沒有條目，則減少所有維度，並返回具有單個元素的張量。

注意：浮點與零的比較是通過精確的浮點相等檢查來完成的。小值是不是出於非零檢查的目的，四舍五入為零。

例如：

x = tf.constant([[0, 1, 0], [1, 1, 0]])
tf.math.count_nonzero(x)  # 3
tf.math.count_nonzero(x, 0)  # [1, 2, 0]
tf.math.count_nonzero(x, 1)  # [1, 2]
tf.math.count_nonzero(x, 1, keepdims=True)  # [[1], [2]]
tf.math.count_nonzero(x, [0, 1])  # 3

注意：字符串與零長度空字符串 "" 進行比較。任何大小大於零的字符串都已被視為非零。

例如：

x = tf.constant(["", "a", "  ", "b", ""])
tf.math.count_nonzero(x) # 3, with "a", "  ", and "b" as nonzero strings.

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.math.count_nonzero。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。