对于全连接层:
nn.Linear(in_features,out_features)
对于输入的是二维的张量来说(一般为(batch size,size)),in_features表示输入的张量的列,out_features表示经过全连接层输出的张量的列,经过全连接层后输出的张量的行不变。
对于卷积层:
输入特征就是通道那个方向上的特征数量。
对于全连接层:
nn.Linear(in_features,out_features)
对于输入的是二维的张量来说(一般为(batch size,size)),in_features表示输入的张量的列,out_features表示经过全连接层输出的张量的列,经过全连接层后输出的张量的行不变。
对于卷积层:
输入特征就是通道那个方向上的特征数量。