图像卷积

第三题,矩阵转置运算后取主对角线之和即可

本身就是借助torch的框架,计算梯度没啥问题,但是自己定义的网络,只能计算2维矩阵