以下の記事を参考に、オートエンコーダ（自己符号化器）による類似画像検索を行うノートブックを作成しました。

https://medium.com/pytorch/image-similarity-search-in-pytorch-1a744cf3469

ノートブックは以下です。

https://colab.research.google.com/github/nakamura196/ndl_ocr/blob/main/Image_Similarity_Search_in_PyTorch.ipynb

「ランタイム」 > 「すべてのセルを実行」 で動作を確認することができます。

参考になりましたら幸いです。