Get Input Dim

Get Input Dim