如何将Pytorch的NLLLoss与计数数据一起使用？

Question

我有一个模型，我试图用整数数据进行预测，其中预期输出的示例为[40]。如何使用nn.PoissonNLLLoss做到这一点？设置log_input = False会使我的损失逐渐降低到0以下。