output_dim_for function

Network Output Dimension