in maddpg/common/distributions.py [0:0]
def mode(self): x = [] for i in range(len(self.categoricals)): x.append(self.low[i] + self.categoricals[i].mode()) return tf.concat(x, axis=-1)