def _MutualInformationTransformAccumulate(pcol, compute_weighted=True):  # pylint: disable=invalid-name
  """Accumulates information needed for mutual information computation."""
  return (pcol | 'VocabCountPerLabelPerTokenAccumulate' >> beam.CombinePerKey(
      _WeightedMeanCombineFn(
          output_shape=(None,), compute_weighted=compute_weighted)))
Accumulates information needed for mutual information computation.
tensorflow_transform/beam/analyzer_impls.py