深度学习模型怎么在最后输出之前加噪声呢？

用这种方法以后 会报错呢

但是如果放在cpu上跑的话就能跑 放在cuda上跑就会出现上述错误 求解答