tf.stop_gradient di pytorch

x.detach()
Lazy Lizard