Python 如何计算一个热张量

我知道我可以使用以下命令将张量转换为热张量：

one\u hot\u labels=tf.one\u hot（labels，depth=3）

现在我想计算一下在

one\u hot\u标签中有多少类0、类1和类2。最简单的计算方法是什么

例如：

输入：

one_hot_labels = [[1,0,0],[1,0,0],[0,0,1]]
one_hot_labels.count([1,0,0]) # something like this command

输出：

2

像这样的东西应该适合你：

one_hot_labels = np.array([[1,0,0],[1,0,0],[0,0,1]])
count_label = tf.reduce_sum(one_hot_labels, axis=0)
sess = tf.Session()
sess.run(count_label)
# array([2, 0, 1])

例如，现在您可以执行以下操作：

count_label = tf.reduce_sum(one_hot_labels, axis=0)[0]
# 2