與gather相對應的逆操作是scatter_,gather把資料從input. 中按index取出,而scatter_是把取出的數據再放回去。注. 意scatter_函數是inplace操作。 out = input.gather(dim ...
確定! 回上一頁