I believe it's FA.shape + (3, 3)