pytorch实现手动滑窗,卷积(利用unfold,fold操作)
import torch
from torch.nn import functional as f
inp = torch.randn(2, 3, 4, 5)
w = torch.randn(2, 3, 3, 3)
inp_unf = torch.nn.functional.unfold(inp, (3, 3)) # [2,27,6]
out_unf = inp_unf.transpose(1, 2).matmul(w.view(w.size(0), -1).t()).transpose(1, 2)
o.
原创
2020-08-07 20:56:01 ·
3305 阅读 ·
3 评论