摘要: 1. torch.nn.Linear PyTorch 中的 nn.linear() 是用于设置网络中的全连接层的,需要注意的是全连接层的输入与输出都是二维张量,一般形状为 [batch_size, size]。 """ in_features: 指的是输入矩阵的列数,即输入二维张量形状 [batch 阅读全文
posted @ 2020-11-28 21:21 _yanghh 阅读(3346) 评论(0) 推荐(1) 编辑