Useful when you have been checking several media using the media viewer navigation open from the chat history and you need to check a particular media in its original context.
The Android and Desktop apps already have this option
Add a three dot button at the top-right corner with the 'View in chat', 'Save Photo' and 'Delete' options (similar to it's already for videos).
A 'View in Chat' / 'Go to Message' button when holding on the media from the media viewer. The UI would be the same than the used when selecting text.