Update return introduction (#11976)

Commit

4 years ago

Update return introduction (#11976) Make it clear that the `forward` method now returns a dict instead of tuple. Fix style