[Iamnotthatbad]

Your example of the commenting doesn't seem like "well-commented code" to me.

Just re-stating what is in code in plain language isn't helpful and is kind of annoying to read.

Can you explain why you think it is helpful?

[sualehasif]

The commenting was for the model to learn what the types were so it could translate it to gather. Doing that one shot is surprising to me because it is somewhat hard to get pytorch tensors to work out to the correct shapes.