摘要:
scatter_add是torch_scatter库中的一个函数,用于对输入张量进行聚合操作,并将聚合结果累加到指定位置上。 具体来说,scatter_add函数的使用方法如下: from torch_scatter import scatter_add # 定义输入张量 input_tensor 阅读全文
posted @ 2023-09-26 20:36
jasonzhangxianrong
阅读(316)
评论(0)
推荐(0)
浙公网安备 33010602011771号