ox_xo007的博客

原创 关于tensor数据结构卷积试验

这里要注意c作为array的维数，否则在作卷积的时候会出现维数差异；学了很多神经网络模型，但是只知道大体框架，具体的实现细节在复制粘贴中也无法明晰。因为随机生成的是一个二维的tensor，所以使用Conv2d来进行卷积处理。收获：通过试验卷积具体数值对卷积更熟悉了，以后看神经网络的代码会有帮助。以下自定义一个内核用于计算网络输入的通道数为1，输出通道数也为1的内核。然后为了试验卷积随机生成一个稍微大些的array转化为tensor。故以下试用tensor，并观察其内部在卷积时的变化。