def _inner_logits(na, emb, qry):
    """Returns logits computed using inner product of qry and embeddings."""
    na_logit = tf.matmul(qry, na, transpose_b=True)
    logits = tf.reduce_sum(tf.expand_dims(qry, 1) * emb[:, 1:, :], 2)
    return tf.concat([na_logit, logits], axis=1)
Returns logits computed using inner product of qry and embeddings.
language/labs/drkit/run_dualencoder_lsf.py