torch.multinomial报错CUDA error: device-side assert triggered

Question

在用pytorch搭建强化学习模型时，想使用torch.multinomial选择动作，但是前两个batch可以正常运行，第三个batch就没办法运行，错误显示:RuntimeError: CUDA error: device-side assert triggered
按照网上教程已经排查了输入数据中没有Nan，但是有0，还有的数值比较小，不知道是不是这个原因。哪位同学遇到过类似的问题可以解答一下，万分感激

#报错行
action = actions_value.multinomial(1)
#传入的action_values
y_pred_nodes.exp()[:, 0, :]