deepof.model_utils.get_hard_counts

deepof.model_utils.get_hard_counts(soft_counts: Tensor)

Compute hard counts per cluster in a differentiable way.

Parameters:

soft_counts (tf.Tensor) – soft counts per cluster