Extract Tensor Values using iterator

Question

I\'m trying to extract the values of label for each image of the dataset. My idea is to split train and test :

# Split into training and test parts
test_datas