Thanks for the suggestion but it wasn’t for a Conv2d layer.
I was trying to place some images on the tensorboard!
The solution in my case was to remove the dodgy 4th colour channel by
T = T[ :, :3, :, : ]