in maddpg/common/distributions.py [0:0]
def logp(self, x): return tf.add_n([p.logp(px) for p, px in zip(self.categoricals, tf.unstack(x - self.low, axis=len(x.get_shape()) - 1))])