Created by: gangliao
@zchen0211 In scatter_test and gather_test, tensor didn't free memory at the end of the program