Pytorch模型如何查看每层输入维度输出维度

NoSuchKey