Added ports to avoid issues with older versions of sklearn

Added a port for train_test_split after some users reported check_arrays problems in certain versions of scikit-learn. Other utilities that need compatibility fixes due to 4D/2D tensor problems can live here. Merging! 

info cc @eickenberg