矩阵 $A$ 应该是 $H \times W$ 的吧。