pytorch里一些函数的说明记录

Posted 猫薄荷喂狗

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch里一些函数的说明记录相关的知识,希望对你有一定的参考价值。

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里一些函数的说明记录的主要内容,如果未能解决你的问题,请参考以下文章