class_docstr: Mixture of distributions with additive i.i.d. uniform noise.
class_docstr_tok: ["Mixture","of","distributions","with","additive","i.i.d",".","uniform","noise","."]
code: def _quantization_offset(self):
      # Picks the "peakiest" of the component quantization offsets.
      offsets = helpers.quantization_offset(
          self.components_distribution
      )
      rank = self.batch_shape.rank
      transposed_offsets = tf.transpose(
          offsets, [rank] + list(range(rank))
      )
      component = tf.argmax(self.log_prob(transposed_offsets), axis=0)
      return tf.gather(offsets, component, axis=-1, batch_dims=rank)
  
imports: import tensorflow
  import tensorflow_probability
  from tensorflow_compression.python.distributions import helpers
label_desc: Use image node_id 7 for calling the NoisyMixtureSameFamily obj's underlying member method code with example usage: obj._quantization_offset() and returns: tf