From Torch Geometric.data Import Data