tf.stop_gradient w Pytorch

x.detach()
Lazy Lizard