I tested, this works :
from torchvision.utils import save_image
import torch
import torchvision

rand_tensor= torch.rand(64, 3,28,28) 

img1 = rand_tensor[0]
# img1 = img1.numpy() # TypeError: tensor or list of tensors expected, got <class 'numpy.ndarray'>
save_image(img1, 'img1.png')