torch.flatten()函数经常用于写分类神经网络的时候,比如《观察torchvision自带的alexnet模型》,经过最后一个卷积层之后,一般会再接一个自适应的池化层,输出 ...
確定! 回上一頁