Update test_example_NN_gpu.m