softmax weights init test now calls set_input_space