so i’m training a 3D Unet where my input size is (1, 1, 64, 64, 64) and my output is (1, 2, 64, 64, 64). How do I get rid of one of the channels for the output so that i can compare the two?
I want my tensor to go from (1, 2, 64, 64, 64) to (1, 1, 64, 64, 64) by removing one of the channels.