move torch.eq out of loop
Codecov / codecov/patch
succeeded
Feb 18, 2024 in 0s
100.00% of diff hit (target 74.93%)
View this Pull Request on Codecov
100.00% of diff hit (target 74.93%)
Loading