def get_output_from_model()

in svhn_based_exp/utils.py [0:0]

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

def get_output_from_model(inputs, net, repre, model, device):
    x = get_repre(torch.flatten(net(inputs), 1), repre, device)
    return x