Implemented getExpectedOutputDim()