首页 > 编程知识 正文

pytorch dropout,pytorch 入门

时间:2023-05-03 06:35:27 阅读:143839 作者:2856

pytorch中的storage指的是连续的内存块,tensor在映射到storage的视图中,将单个内存区域映射到了n维空间视图。

size是tensor维,storage offset是storage中数据的索引,例如stride是storage中对应于tensor的相邻维度间第一个索引的跨度,也叫步长

上图为storage,其对应的tensor ([ 3.0,1.0,2.0 ]、[ 4.0,1.0,f] )如下图所示。

那么,这里的stride=(3,1),因为从第一行的第一个索引到第二行第一个索引跨度是3,从第一列到第二列的跨度是1,stride是元组,stride[0]=3,stride[1]=1; 这里的size是torch.size ([ 2,3 ] ); 根据tenser的索引I,j表示用于查找storage中相应索引的表达式为storage_offset + stride[0] * i + stride[1] * j,从storage的开头开始查找,因此storage_offset=0。

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。