Python tf.compat.v1.count_nonzero用法及代碼示例

計算張量維度上非零元素的數量。 (不推薦使用的參數)(不推薦使用的參數)

用法

tf.compat.v1.count_nonzero(
    input_tensor=None, axis=None, keepdims=None, dtype=tf.dtypes.int64, name=None,
    reduction_indices=None, keep_dims=None, input=None
)

警告：不推薦使用某些參數：(keep_dims)。它們將在未來的版本中被刪除。更新說明：keep_dims 已棄用，請改用 keepdims

警告：不推薦使用某些參數：(reduction_indices)。它們將在未來的版本中被刪除。更新說明：reduction_indices 已棄用，請改用軸

沿 axis 中給定的尺寸減少 input_tensor 。除非 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.