请教一个PyTorch问题

BarryBarry  

2017-11-22

请问各位,用pytorch的nn.conv2d()怎么实现由输入的shape是(1,1,5,5),卷积核shape是(2, 5),经过卷积产生一个(1,1,5,1)的输出?

关注问题 3人已关注
 写回答 3人已回答