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

pytorch中gather函数的理解。

2020-05-11pytorch自定义求导

将函数元素应用于 Pytorch CUDA 张量的并行方式

pytorch安装记录

pytorch之transforms.Compose()函数理解

仿真基本功PyTorch中torch.Tensor变形的相关函数使用方法及说明