1. 程式人生 > >pytorch裏一些函數的說明記錄

pytorch裏一些函數的說明記錄

first ide slice bsp mesi blog body 上進 int

TH_API void THTensor_(narrow)(THTensor *self, THTensor *src, int dimension, long firstIndex, long size);

從src裏找到第dimension個維度,從這個維度的firstIndex開始,連續取size個子tensor

self復制於src,通過修改self->storageOffset和self->size[dimension]來維持視圖的正確。

TH_API void THTensor_(select)(THTensor *self, THTensor *src, int dimension, long sliceIndex);

去src裏的第dimension維度裏的第sliceIndex個子tensor, 結果在self上進行修改

self通過前移size、stride,修改nDimesion來維持視圖正確。

調用了THTensor_(narrow)

pytorch裏一些函數的說明記錄