深度学习（PyTorch）——flatten函数的用法及其与reshape函数的区别