- 博客(7)
- 收藏
- 关注
原创 010 CNN基础
这一行代码是用来获取当前批次(batch)的数据量。在 PyTorch 中,张量。外的其他维度,即这里的 320。这样就可以将展平后的向量输入到全连接层中了。这一行中用于将卷积层的输出展平成一个一维向量,其中。表示一批次中包含的样本数量。
2024-08-03 15:21:08
122
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人