Compute the similarity between two input tensors. Make sure to compute the softmax if softmax is True.
softmax
SimilarityModule.forward()
dgs.models.similarity.similarity.SimilarityModule.configure_torch_module
dgs.models.similarity.similarity.SimilarityModule.get_data