转置卷积

卷积不会增大输入的高宽，通常要么不变、要么减半；

转置卷积则可以用来增大输入高宽。

$$Y[i:i+h,j:j+w]+=X[i,j]\cdot K$$

如果卷积将输入从$(h,w)$变成了$(h^{\prime },w^{\prime })$，同样超参数的转置卷积则从$(h^{\prime },w^{\prime })$变成$(h,w)$。

代码实现

import torch
from torch import nndef trans_conv(X, K):h, w = K.shapeY = torch.zeros((X.shape[0] + h - 1, X.shape[1] + w - 1))for i in range(X.shape[0]):for j in range(X.shape[1]):Y[i:i + h, j:j + w] += X[i, j] * Kreturn Y