scatter_(input, dim, index, src)用法

scatter_(input, dim, index, src)用法 0 表示按行填充 index中的位置對應x的位置取出用來填充的值,index的值的值對應input的要填充那個位置的行號,index的列號對應要填充input的列號例如index[0][2]----src[0][2]=0.4497 對應input[2][2] = 0.4497 ​同理得到按列填充一般用來進行onehot的處理 吐
相關文章
相關標籤/搜索