[primTorch] Implement NLL loss reference (#81128)
Add Reference:
- nll_loss
Depends on:
- expand https://github.com/pytorch/pytorch/pull/79820
- advance indexing
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81128
Approved by: https://github.com/mruberry