tf.stop_gradient in pytorch

x.detach()
Lazy Lizard