We can extend the idea of this metric to measure how similar the original image and edited version are. To do that, we can just do`F.cosine_similarity(img_feat_two, img_feat_one)`. For these kinds of edits, we would still want the primary semantics of the images to be preserved as much as possible, i.e., a high similarity score.