Minor change to _transform_input method in NNInterface
I have changed the implementation of the _transform_input method in the NNInterface class slightly, so it also accepts 4-dimensional inputs.
I have changed the implementation of the _transform_input method in the NNInterface class slightly, so it also accepts 4-dimensional inputs.