5-jsma-pytorch.ipynb代码在cuda上运行代码修改

[igeng]

兜哥您好，我在复现您的code/5-jsma-pytorch.ipynb代码时候，发现循环训练中最后一行
img.data[idx]=np.clip(img.data[idx], min_, max_)
该行代码在cuda上运行需要修改，img在cuda上运行的时候，因为img是cuda tensor，需要转化为cpu tensor，再进行numpy运算。
img.data[idx]=np.clip(img.data[idx].cpu(), min_, max_)