1. Fix the bug in SubMe function which cannot handle the scalar tensor and broadcast case. 2. Clean the codes.
