get_hard_counts()
Compute hard counts per cluster in a differentiable way.
soft_counts (tf.Tensor) – soft counts per cluster