最完整的PyTorch数据科学家指南(2)

本篇文章继续为同学们讲解神经网络的卷积层部分。 因此,Conv2d图层需要使用Cin通道将高度为H且宽度为W的图像作为输入 。现在,对于卷积网络中的第一层,的数量in_channels将为3(RGB),并且out_channels用户可以定义数量。kernel_size大多采用3×3是,并且stride通常使用为1。 为了检查一个我不太了解的新层,我通常尝试查看该层的输入和输出,如下所示,在该层我
相关文章
相关标签/搜索