in mico/utils/net_utils.py [0:0]
def forward(self): """Return the approximated distribution. """ logits = self.theta.weight probability_vector = self.softmax(logits) return probability_vector