tf.stop_gradient в pytorch

x.detach()
Lazy Lizard