Train the neural network with negative lokelihood for regression

Question

I am trying to train the simple feedforward neural network using the negative-likelihood function to estimate the uncertainty for the regression task. My neural network outp