记录一次自定义损失导致的RuntimeError: CUDA error: device-side assert triggered

C:\cb\pytorch_1000000000000\work\aten\src\ATen\native\cuda\Loss.cu:115: block: [0,0,0], thread: [15,0,0] Assertion `input_val >= zero && input_val <= one` failed

RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

完整错误如下出现这个问题是因为我自定义了一个损失,这个损失用到了开方操作,开方内部出现了极小值甚至负值,导致损失返回nan造成的.我把开方内部加上绝对值再加一个1e-9就没问题了.