Thank you! That’s work, but I realized the “one_hot_labels” at the end of the loop has a size [2,10], it’s possible that has size [2,10] inside the loop?
If I want to concatenate more than two images, which parts of inputs and one_hot_labels should be changed?