Device-to-device (D2D) transmission has become an essential form of wireless communication due to the rise of 5G and Internet of Things (IoT). Unfortunately, most currently available techniques for allocating resources are extremely time-consuming or computationally expensive. Graph neural networks (GNNs) have recently been proposed as a way to improve the efficacy of many network-related tasks. To tackle this issue, we propose a GNN-based method in a supervised manner. We denote heterogeneous networks as directed graphs and model various communication links as nodes and interference between links as edges. The proposed algorithm has two phases: (1) Offline supervised learning obtains the optimal solution through the particle swarm optimization (PSO) algorithm and uses the solution as a sample label to train GNN; (2) Online prediction by well-trained GNN. The simulation results indicate that the proposed method outperforms the state-of-the-art GNN-based benchmarks and achieves substantial speedups over conventional benchmarks.