PyTorch的nn.Linear()
是用于设置网络中的全连接层的。
nn.Linear(in_features,out_features
)
相当于一个输入为[batch_size, in_features]
的张量变换成了[batch_size, out_features]
的输出张量。
nn.Linear()
是用于设置网络中的全连接层的。in_features,out_features
)相当于一个输入为[batch_size, in_features]
的张量变换成了[batch_size, out_features]
的输出张量。