Replace the original node IDs with shuffled node IDs in the edge list by using tensor operations instead of python dict.

[jimjlu99]

🔨Work Item

IMPORTANT:

• This template is only for dev team to track project progress. For feature request or bug report, please use the corresponding issue templates.
• DO NOT create a new work item if the purpose is to fix an existing issue or feature request. We will directly use the issue in the project tracker.

Project tracker: https://github.com/orgs/dmlc/projects/2

Description

Replace the original node IDs with shuffled node IDs in the edge list by using tensor operations instead of python dict.

Depending work items or issues