Shampoo对每层权重矩阵分别维护行、列方向的协方差矩阵并做Kronecker因子分解。