import torch
# from torch_geometric.loader import DataLoader
from torch_geometric.data import Data, DataLoader,InMemoryDataset
class PairData(Data):
def __init__(self,w=None, edge_index_s=None, x_s=None, edge_index_t=None, x_t=None):
super().__init__()
self.x_s = x_s
self.x_t = x_t
self.edge_index_s=edge_index_s
self.edge_index_t=edge_index_t
self.w=w
def __inc__
pairdata使用 torch geometric
最新推荐文章于 2023-10-01 19:45:43 发布
关键词由CSDN通过智能技术生成