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
```