一个微不足道的评论错误

我是一名学生,最近我在研究这个项目时,发现了一个非常模糊的注释错误。在 models.py 中的 unpatchify 函数中, imgs 被标注为 (N,H,W,C),但其实应该是 (N,C,H,W)。