add set_input_dim on the pycaffe