pytorch相关

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qxqsunshine/article/details/84000065

1、autograd自动微分

所以这autograd只对输出值为标量的有用,因为损失函数的输出是一个标量。

variable 是tensor的外包装,variable类型变量的data属性存储着tensor数据,grad属性存储关于该变量的导师,

博客解说:https://www.cnblogs.com/CATHY-MU/p/7754998.html

2、构造一个小型的CNN

torch.nn只接受mini-batch的输入,也就是说我们输入的时候是必须是好几张图片同时输入。

例如:nn. Conv2d 允许输入4维的Tensor:n个样本 x n个色彩频道 x 高度 x 宽度

猜你喜欢

转载自blog.csdn.net/qxqsunshine/article/details/84000065