为什么模型只能在 [64, 64] 大小上进行训练?

当我尝试使用不等于 [64, 64] 的大小时，会出现以下错误： 文件“/home/PyTorch-VAE/models/vanilla_vae.py”，第 122 行，在 forward 中 文件“/home/PyTorch-VAE/models/vanilla_vae.py”，第 91 行，在 encode 中 文件“/home/anaconda3/envs/py38/lib/python3.8/site-packages/torch/nn/modules/module.py”，第 889 行，在 _call_impl 中 文件“/home/anaconda3/envs/py38/lib/python3.8/site-packages/torch/nn/modules/linear.py”，第 94 行，在 forward 中 文件“/home/anaconda3/envs/py38/lib/python3.8/site-packages/torch/nn/functional.py”，第 1753 行，在 linear 中 运行时错误：mat1 的第 1 维必须与 mat2 的第 0 维匹配