ouput = torch.scatter(input, dim, index, src) # 或者是 input.scatter_(dim, index, src)兩個(gè)方法的功能是相同的,而帶下劃線的 _scatter 方法是將原tensor input 直接修改了,不帶的則會(huì)返回一個(gè)新的tensor output , input 不變。其中dim 決定index 對(duì)應(yīng)值是
www.dbjr.com.cn/python/297906u...htm 2025-6-6