why the input_size of Linear if 2?
why the input_size of Linear if 2?