对于某需要输入到nn.Linear(in_features, out_features)中的tensor[a, b, c, d, e]需要满足:e == in_features
对于某需要输入到nn.Conv2d(in_channels, out_channels)的中的tensor[a, b, c, d]需要满足:b == in_channels
对于某需要输入到nn.Linear(in_features, out_features)中的tensor[a, b, c, d, e]需要满足:e == in_features
对于某需要输入到nn.Conv2d(in_channels, out_channels)的中的tensor[a, b, c, d]需要满足:b == in_channels