how I display the result of the convolution as an image?
out1 = self.cnn6(out1)
        out1 = self.batchnorm6(out1)
        out1 = self.relu(out1)        
        plt.imshow(np.transpose(out1, (1,2,0)), interpolation='nearest')

RuntimeError: Can’t call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.