2023年10月24日
7、生成mask，过滤无效张量
摘要： 1、准备环境 import torch random_tensor = torch.randn(10, 2) print(random_tensor) 2、准备batch索引 from torch_geometric.utils import to_dense_batch, to_dense_adj 阅读全文
posted @ 2023-10-24 22:59 jasonzhangxianrong 阅读(52) 评论(0) 推荐(0) 编辑