关于pytorch中nn.conv中的参数

关于pytorch中nn.conv中的参数 之前一直是用的tensor flow +keras,最近刚转为pytorch,发现torch中的卷积参数和tensorflow中的有点不同,特来记录下,以下是官网中的解释: 和tensorflow不同,in_channels和out_channels分别是输入和输出的通道数,stride步长以及padding填充和keras是一样的,groups 这个参
相关文章
相关标签/搜索