Pytorch 卷積中的 Input Shape用法

先看Pytorch中的卷積 class torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True) 二維卷積層, 輸入的尺度是(N, C_in,H,W),輸出尺度(N,C_out,H_out,W_out)的計算方式   這裏比較奇怪的是這個
相關文章
相關標籤/搜索