add back inputDimensions and outputDimensions to AMatrix