forgot transpose change to SerialDenseWrapper