沿着张量流中给定张量的轴计算模和模数

Question

我有一个二维数组为：

c=np.array([[1, 3, 0, 0, 3],
            [1, 3, 1, 0, 2],
            [1, 3, 1, 2, 2]])

我想沿着轴= 0计算模式和模式计数。因此结果应如下所示：

mode = [1,3,1,0,2], mode-count=[3,3,2,2,2]

我已经在TensorFlow网站上进行了搜索，但是除了找到其他有用的API外，tf.unique_with_counts，需要一维张量。

我不想在数组c的每一列上运行循环，以便将tf.unique_with_counts用于计算模式和模式计数。带有示例的任何建议都是最欢迎的。

Answer 1

TensorFlow Probability具有tfp.stats.count_integers函数，可以使这一过程非常简单：

tfp.stats.count_integers