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中

>>阅读原文<<