I have a filter output from a single layer in the network, how do I then split the output into individual RGB channels to view the filter in each channel separately. I currently have this, but it just looks like the image is coloured:
    weights = weights - weights.min()
    weights = weights / weights.max()
    display = torchvision.utils.make_grid(weights, nrow = 8)
    display = display.permute(1, 2, 0)
    plt.imshow(display[:,:,0], cmap='Reds')
    plt.show()
    plt.imshow(display[:,:,0], cmap='Greens')
    plt.show()
    plt.imshow(display[:,:,0], cmap='Blues')

which gives images: