code: def test_embed_similarity():
      """Test embed similarity."""
      embeds = torch.rand(2, 3)
      similarity = embed_similarity(embeds, embeds)
      assert similarity.shape == (2, 2)
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_embed_similarity() without return types