Actually I’m using 1200 images for training and 400 images for test. Therefore I have created these tensors and saved numpy array data of 1200 images into the train tensor and 400 images into test tensor