tf.stop_gradient en pytorch

x.detach()
Lazy Lizard