def forward()

in mico/utils/net_utils.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

    def forward(self, inputs):
        logits = self.ff(inputs).view(inputs.size(0), 1, self.number_clusters)
        probability_vector = self.softmax(logits)
        return probability_vector