bdusell / semiring-einsum

Generic PyTorch implementation of einsum that supports different semirings
https://bdusell.github.io/semiring-einsum/
MIT License
45 stars 7 forks source link

Change clip_max_values to check for +inf as well. #14

Closed davidweichiang closed 2 years ago

davidweichiang commented 2 years ago

It now uses torch.nan_to_num, so as a by-product, replaces nan with 0, which I believe is harmless here.

davidweichiang commented 2 years ago

Closes #13