Ejemplo n.º 1
0
 def ann(qy, db, db_half_norm, k):
     scores = db_half_norm - math_ops.matmul(qy, db, transpose_b=True)
     return nn_ops.approx_min_k(scores, k)
Ejemplo n.º 2
0
 def ann(db, k):
     return nn_ops.approx_min_k(db, k=k, reduction_dimension=2)
Ejemplo n.º 3
0
 def ann(db, k=10):
     return nn_ops.approx_min_k(db,
                                k,
                                aggregate_to_topk=aggregate_to_topk)