实现简单的straight-through estimator(STE)（pytorch版）

tensor([-0.4608, 0.0240, 0.5585, -0.7694], requires_grad=True)tensor([0., 0. ...