code: def forward(self, idx):
      # embeddings are stored in cpu
      idx = idx.cpu()
      if self.dgl_sparse_emb:
          return self.sparse_emb(idx, device=self.dev_id)
      else:
          return self.sparse_emb(idx).to(self.dev_id)
  
file_path: dgl/examples/pytorch/graphsage/dist/train_dist_transductive.py
imports: import argparse
  import time
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  import torch.optim
  import dgl
  from dgl.distributed import DistEmbedding
  from train_dist import DistSAGE, compute_acc
label_desc: Use image node_id 2 for calling the DistEmb obj's underlying member method code with example usage: obj.forward(idx) and returns: self, self