摘要: 感谢澄清!让我更详细地解释原来代码中的 torch.bmm(coord_diff, coord_diff.transpose(-1, -2)) 与后续改进后的实现之间的区别。 原始代码的含义 coord_diff = coord[row] - coord[col] # [n_edge, 14, 3] 阅读全文
posted @ 2024-10-11 17:21 GraphL 阅读(45) 评论(0) 推荐(0)