code: def test_complex_scoring():
      model = ComplEx(num_nodes=5, num_relations=2, hidden_channels=1)
  
      model.node_emb.weight.data = torch.tensor(
          [
              [2.0],
              [3.0],
              [5.0],
              [1.0],
              [2.0],
          ]
      )
      model.node_emb_im.weight.data = torch.tensor(
          [
              [4.0],
              [1.0],
              [3.0],
              [1.0],
              [2.0],
          ]
      )
      model.rel_emb.weight.data = torch.tensor(
          [
              [2.0],
              [3.0],
          ]
      )
      model.rel_emb_im.weight.data = torch.tensor(
          [
              [3.0],
              [1.0],
          ]
      )
  
      score = model(
          head_index=torch.tensor([1, 3]),
          rel_type=torch.tensor([1, 0]),
          tail_index=torch.tensor([2, 4]),
      )
      assert score.tolist() == [58.0, 8.0]
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_complex_scoring() without return types