[Feature] `zero_grad` and `requires_grad_` - Githubissues

pytorch / tensordict

TensorDict is a pytorch dedicated tensor container.

MIT License

803 stars 65 forks source link

[Feature] `zero_grad` and `requires_grad_` #901

Closed vmoens closed 1 month ago