Single Convolution layer on MNIST dataset

Hi Team,

I’m trying to apply single convolution layer on MNIST dataset, the way it’s been shown in CNN python video. But the plot showing bit different for me, i don’t think it’s plotting correctly. It should be some blurred image not lines like shown below. Please let me know how can I get correct image here.

FirstCNN class is same as described in the video.

class FirstCNN(nn.Module):

def __init__(self):

super(FirstCNN, self).\_\_init\_\_()

self.conv1 = nn.Conv2d(1,16,3)

def forward(self,x):

x = self.conv1(x)

return x

Please use plt.imshow() instead of plt.plot()