Don't keep output of a layer as 2d matrix

[I159]

It has no sense because it never used in not scaled down state and also dot product of 1d matrix by 2d matrix is 1d.