My contribution was implementing the Flatten module for the C++ front-end of PyTorch!
Primary LanguageC++OtherNOASSERTION