我想以最有效的方式计算以下运算：
cc = nn.Conv1d(1,1,3,1,1,bias=False) U = torch.randn(1,1,10,10) V = torch.zeros_like(U) for i in range(10): V[:,:,:,i] = cc(U[:,:,:,i])
换句话说，我想对输入U的列应用一维卷积。当然for循环太慢了，我确信有更有效的方法来解决这个问题。但是，我找不到任何想法来实现这一点。
fwzugrvs1#
您可以使用矩形窗口（即1x3）应用2D卷积：
1x3
conv = nn.Conv2d(1,1, kernel_size=(1,3), padding=1, bias=False)
1条答案
按热度按时间fwzugrvs1#
您可以使用矩形窗口（即
1x3）应用2D卷积：