Fix dimensionality issue when binary classification outputs 1D instead of 2D tensor #3673

Job	Run time
Mypy Type Checks	1m 49s
format	10s
Build + Test (C++) (Release, gcc, g++, 11)	1m 50s
C++ Test Coverage (gets coverage of main branch, currently not main branch because no C++ on main)	1m 51s
Build + Test (C++) (Release, clang, clang++, 14)	1m 50s
Pre-commit Checks	28s
pylint	1m 49s
Build + Test (C++) (Release, clang, clang++, 17, true)	1m 50s
tidy	1m 47s
Build + Test (C++) (Debug, gcc, g++, 11)	1m 51s
unittests	1m 50s
Build + Test (C++) (Debug, gcc, g++, 12)	1m 51s
Build + Test (C++) (Debug, clang, clang++, 14)	1m 52s
Build + Test (C++) (Debug, clang, clang++, 17, true)	1m 50s
Build + Test (C++) (clang, clang++, 14, Tsan)	1m 50s
Build + Test (C++) (clang, clang++, 14, Asan)	1m 50s
Comment Coverage Results	1s
dockerized-unittests	1s
integrationtests-debug	1s
integrationtests-asan	1s
integrationtests-release	1s
integrationtests-tsan	1s
	26m 24s

Provide feedback