Python – tensorflow.math.count_nonzero()
TensorFlow是谷歌设计的开源Python库,用于开发机器学习模型和深度学习神经网络。
count_nonzero()用于计算张量中非零元素的数量。
语法: tf.math.count_nonzero( input, axis, keepdim, dtype, name)
参数:
- input:它是一个需要被减少的张量。
- axis(可选):它定义了需要减少输入的轴。允许的范围是[-rank(input), rank(input)]。如果没有给定值,则默认为无,即输入将沿所有轴被减少。
- keepdim(可选):如果为真,将保留长度为1的缩小尺寸。
- dtype(可选):它定义了输出的dtype。默认是int32。
- name(可选):它定义了该操作的名称。
返回值:
它返回一个包含非零值数量的张量。
示例 1:
输出:
示例2:当输入张量为字符串类型时,“”被认为是空字符串。“”为非零。
输出: