torch.nn.Conv2d()函数详解

import torch x = torch.randn(2,1,7,3) conv = torch.nn.Conv2d(1,8,(2,3)) res = conv(x) print(res.shape) # shape = (2, 8, 6, 1) 输入： x [ batch_size, channels, height_1, width_1 ] batch_size 一个batch中

>>阅读原文<<