RElbers/info-nce-pytorch

reduction mode

Closed this issue · 1 comments

Thx for your implementation!
Have you ever considered about setting the reduction mode, it seems that can be easily modified.

I have added the reduction mode as a parameter.