for i in train_loader: images.shape = [64,3,28,28] I can save images but 64 images are drawed. How can I save only one image from them?
確定! 回上一頁