python
打了小松鼠
这个作者很懒,什么都没留下…
展开
-
pytorch.stack 函数的详解
首先不要用行列式或在大脑中建立3D空间这种思维方式来分析这个函数。 stack 函数返回的结果会新增一个维度。 stack函数指定的dim参数,就是新增维度的(下标)位置。 例如:a=[i][j], b=[i][j],注意这里a中的i、j和b中的i、j必须相同否则会报错。 在stack函数中指定dim=0,则新维度代表第一个下标。 c = stack((a,b), dim=0) 则 c[0][i]...原创 2020-04-02 15:15:07 · 3156 阅读 · 1 评论 -
pytorch gather函数理解
pytorch gather函数理解 注释中的说法如下 out[i][j][k] = tensor[index[i][j][k]][j][k] # dim=0 out[i][j][k] = tensor[i][index[i][j][k]][k] # dim=1 out[i][j][k] = tensor[i][j][index[i][j][k]] # dim=2 看起来有点不好理解,因此做了如下...原创 2020-03-23 00:03:42 · 213 阅读 · 0 评论 -
Zip 对象----构造函数返回生成器对象
Zip 对象----构造函数返回生成器对象 Zip对象不能直接通过下标操作数据,但可以通过next函数来取数据。 Zip对象返回的是一个生成器,可以通过下面的代码理解生成器。 在for里面是用yield,这就是生成器的关键。通常情况下我们习惯像genseq2中那样返回一个list,这个list对象会一直存在于内存之中。而通过生成器可以不用return 将需要return出来的元素用yield修饰。...原创 2019-11-10 22:44:37 · 400 阅读 · 0 评论