Hi
I have the below code but i don’t know where is wrong that i get error
trainset = torchvision.datasets.MNIST(root ='./data',
                                        download=True,
                                        transform=transforms.Compose(
        [transforms.ToTensor(), transforms.Lambda(lambda x: x * 64)]
    ))
x= trainset[5] 
plt.imshow(x, cmap='gray')

here is the error:
ValueError: only one element tensors can be converted to Python scalars