how to iterate over tensor in tensorflow