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, :, : ]
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, :, : ]