Translate PyTorch to TensorFlow - Gradients become `None`

(topic deleted by author)