code: def check_all_set_all_get_emb(device, init_emb):
      num_embs = init_emb.shape[0]
      emb_dim = init_emb.shape[1]
      dgl_emb = NodeEmbedding(num_embs, emb_dim, "test", device=device)
      dgl_emb.all_set_embedding(init_emb)
  
      out_emb = dgl_emb.all_get_embedding()
      assert F.allclose(init_emb, out_emb)
  
file_path: dgl/tests/python/pytorch/nn/test_sparse_emb.py
imports: import multiprocessing
  import os
  import unittest
  import backend
  import pytest
  import torch
  from dgl.nn import NodeEmbedding
  from dgl.optim import SparseAdam
label_desc: Use image node_id 2 for calling a global function with example usage: check_all_set_all_get_emb(device, init_emb) without return types