2022年11月3日
摘要: conv = torch.nn.Conv2d(180, 2, (3, 3), stride=1, padding=1) print(conv.weight.data.shape) #卷积核权重是一个四维张量,第一个维度表示卷积核个数,第二个维度表示每个卷积核的层数,第三和第四个维度表示卷积核的宽 输 阅读全文
posted @ 2022-11-03 15:58 沈香茶 阅读(185) 评论(0) 推荐(0)