Pytorch clone tensor gradient