search

shenweichen / DeepMatch

A deep matching model library for recommendations & advertising. It's easy to train models and to export representation vectors which can be used for ANN search.
https://deepmatch.readthedocs.io/en/latest/
Apache License 2.0
2.22k stars 530 forks source link

enhance inner_product function #97

Open SunnyCat2013 opened 1 year ago

SunnyCat2013 commented 1 year ago

update parameters with axis and keepdims