2021
06-22
06-22
Pytorch实现全连接层的操作
全连接神经网络(FC)全连接神经网络是一种最基本的神经网络结构,英文为FullConnection,所以一般简称FC。FC的准则很简单:神经网络中除输入层之外的每个节点都和上一层的所有节点有连接。以上一次的MNIST为例importtorchimporttorch.utils.datafromtorchimportoptimfromtorchvisionimportdatasetsfromtorchvision.transformsimporttransformsimporttorch.nn.functionalasFbatch_size=200learning_rate=0.001e...
继续阅读 >