5.7.3 使用PyTorch实现CW