def get_answer_embedding()

in src/scripts/ffn_inference.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

def get_answer_embedding(model, input_emb, output_dim):
    res = np.zeros((input_emb.shape[0], output_dim))
    for i, input in enumerate(input_emb):
        input_tn = torch.from_numpy(input).double()
        output = model(input_tn).detach().numpy()
        res[i] = output
    return res