def call(self, args, mask=None):
    return self.multiplexer(args, self.output_dim, self.nb_ctrl_sig)
def call(self, args, mask=None):
    return self.multiplexer(args, self.output_dim, self.nb_ctrl_sig)