提交 ee08da1f 编写于 作者: R rusty1s

fix

上级 288cfd44
......@@ -131,7 +131,7 @@ relabel_one_hop_cpu(torch::Tensor rowptr, torch::Tensor col,
}
}
if (bipartite)
if (!bipartite)
out_rowptr = torch::cat(
{out_rowptr, torch::full({(int64_t)n_ids.size()}, out_col.numel(),
rowptr.options())});
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册