标签： python graph pytorch
如果我有一个 data 对象，其中还包含 .train_mask 和 .test_mask 张量，我如何使用它们将数据集拆分为两个数据对象，一个 trainset和一个testset？
data
.train_mask
.test_mask
trainset
testset