Comment by Y_Y

Seconded. It feels like cheating but it takes out a lot of boilerplate.

Overall pytorch is a product of suspiciously high quality. I would say "except for libtorch" but it looks like they might have remedied that in this release.