Torch7 Tensor切片總結


1.narrow(k,m,n)

這個函數是選中第k維的從m行開始,供選中n行
2.sub(dim1s,dim1e[,dim2s,dim2e,..,dim4s,dim4e])
功能最強大,可以切任意的一個字塊,不過參數比較多,制定每一維的從開始到到結束的index
3.select(dim, index)
這個最實用於選擇column,因為選中第3列用大括號的表達為A[{{},{3}}] 太麻煩了,還不如直接A:select(2,3)來的簡單有效
4.index(dim,index)
 
離散選擇行列用這個,所謂離散的意思就是我想選中第一行和第4行,這個前面的函數都不能直接進行操作。
index卻可以,比如A:index(1,torch.LongTensor{1,4})
需要注意的是這個函數返回新的Tensor不是在原來的Tensor的存儲空間上做操作了!!!!
 
基於index可以做indexCopy, indexFill, indexAdd等操作,但是這個是在原始storage上做的。
 
gather這個函數
 
The gather operation
maskSelected可以用來做dropout這種功能,通過制定一個mask (0,1)矩陣來決定哪個要哪個不要。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM