Torch7 Tensor切片總結

2022-05-04 04:33:10 字數 791 閱讀 8809

1.narrow(k,m,n)

這個函式是選中第k維的從m行開始,供選中n行

2.sub(dim1s,dim1e[,dim2s,dim2e,..,dim4s,dim4e])

功能最強大,可以切任意的乙個字塊,不過引數比較多,制定每一維的從開始到到結束的index

3.select(dim, index)

這個最實用於選擇column,因為選中第3列用大括號的表達為a[,}] 太麻煩了,還不如直接a:select(2,3)來的簡單有效

4.index(dim,index)

離散選擇行列用這個,所謂離散的意思就是我想選中第一行和第4行,這個前面的函式都不能直接進行操作。

index卻可以,比如a:index(1,torch.longtensor)

需要注意的是這個函式返回新的tensor不是在原來的tensor的儲存空間上做操作了!!!!

基於index可以做indexcopy, indexfill, indexadd等操作,但是這個是在原始storage上做的。

gather這個函式

maskselected可以用來做dropout這種功能,通過制定乙個mask (0,1)矩陣來決定哪個要哪個不要。

torch入門筆記8 tensor

這是自己定義tensor的值輸入10個整型值,列印price,注意這時它只是乙個有10資料的tensor。對比這兩張,第一張是有10個資料的tensor,第二張是10x1的矩陣,也可以成為列向量 神經網路經常用到 是呼叫了reshape 方法轉換的,它們看著一樣本質上是不一樣的!這時再次呼叫resh...

Torch7模型訓練

torch7搭建卷積神經網路詳細教程 已經詳細的介紹啦module模組,這裡再次基礎上再給出一些上container transfer functions layers和 layers模組的理解。並在後面給出一些簡單的模型訓練方法。下述程式在itorch qtconsole下執行。主要有四個函式 詳...

Torch7 out of memory 解決方法

造成out of memory 的主要原因是lua預設的編譯器luajit的記憶體限制為2gb,解決的方法是用lua代替luajit,介紹了替換方法,具體操作如下 1.解除安裝之間安裝的torch 1.1 進入torch安裝路徑 clean.sh 1.2 刪除安裝的檔案 rm rf install....