I have a numpy ndarray train_data of length 200, where every row is another ndarray of length 10304.
train_data
200
10304
However when I print