Can't find implementation of nll_loss_nd

PydevC · July 25, 2025, 9:11am

I wanted to checkout the for implementation of torch.nn.functional.nll_loss()
I came across this line where nll_loss was implemented. Looking into the code at the location torch._c._nn.pyi I was unable to find code related to losses but had no luck.

github.com/pytorch/pytorch

torch/nn/functional.py

134179474


      
          return torch._C._nn.nll_loss_nd(
              input, target, weight, _Reduction.get_enum(reduction), ignore_index
          )

Does anyone where I can find the implementation of the function mentioned above.
Thanks in advance

TheRecklessDoctor · August 22, 2025, 10:18am

Hi, the function nll_loss_nd should be defined in this file:

github.com/pytorch/pytorch

torch/onnx/symbolic_opset12.py

main


      
          
          
          @_onnx_symbolic("aten::nll_loss2d")
          def nll_loss2d(
              g: jit_utils.GraphContext, self, target, weight, reduction, ignore_index
          ):
              return nll_loss(g, self, target, weight, reduction, ignore_index)
          
          
          @_onnx_symbolic("aten::nll_loss_nd")
          def nll_loss_nd(
              g: jit_utils.GraphContext, self, target, weight, reduction, ignore_index
          ):
              return nll_loss(g, self, target, weight, reduction, ignore_index)
          
          
          @_onnx_symbolic("aten::cross_entropy_loss")
          def cross_entropy_loss(
              g: jit_utils.GraphContext,
              self,
              target,

Topic		Replies	Views
Torch.nn September Update frontend API	0	813	September 16, 2021
NNC walkthrough: how PyTorch ops get fused nnc	10	7377	November 3, 2021
NNC Per-Operator Benchmarks (on CPU) nnc	5	1061	January 27, 2021
About the nnc category nnc	0	558	January 22, 2021
Torch.nn H2 2021 Lookback and H1 2022 Lookahead frontend API	0	972	January 14, 2022

Can't find implementation of nll_loss_nd

Related topics