[tensor_trainer] Add a function that generates dummy data

[tensor_trainer] Add a function that generates dummy data #342