tf.stop_gradient à pytorch

x.detach()
Lazy Lizard