本文共 281 字,大约阅读时间需要 1 分钟。
def corr2d_multi_in_out_1x1(X, K): c_i, h, w = X.shape c_o = K.shape[0] X = X.reshape((c_i, h * w)) K = K.reshape((c_o, c_i)) Y = torch.matmul(K, X) # Matrix multiplication in the fully-connected layer return Y.reshape((c_o, h, w))
转载地址:http://qrzai.baihongyu.com/