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

pytorch / tensordict

TensorDict is a pytorch dedicated tensor container.

MIT License

832 stars 74 forks source link

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

Closed vmoens closed 3 months ago