Solved by little_adventurer in post #3 
              
              
                If anyone ever encounters a similar problem, the solution seems to be simply slicing the tensor and removing the 4th colour channel. 
new_tensor = T[ :, :3, :, : ]