遨游大海的猪

一句话: Module.modules(): 采用了深度优先遍历的方式,如果想递归的遍历所有的模块,包括自己,使用 Module.modules(). Module.children(): 如果只想遍历儿子这一代模块, 使用 Module.children(). 详细的解释来源如下: 链接: https://discuss.pytorch.org/t/module-children-vs-...

torch.stack() 含义:在指定的维度对元素进行叠加 例子: 使用torch生成如下的三个tensor: a, b, c 在指定的维度 0 对a, b, c进行叠加: 在指定的维度 1 对a, b, c进行叠加: 在指定的维度 2 对a, b, c进行叠加: 注意: 对于维度 0, 1, 2应该从外往里面看进去. 在指定的维度对tensor进行叠加时, 应该把...

Variable和Tensor本质上一样,不过Variable会放入一个计算图里面,然后进行前向传播,反向传播以及自动求导. 一个Variable里面包含着三个属性,data,grad和creator,其中creator表示得到这个Variable的操作,比如乘法或者加法等,grad表示方向传播的梯度,data表示取出这个Variable里面的数据. ...