I have a tensor of size [B, 64, 256, 384]. I want to stretch it softly along the channel axis so it becomes [B, 256, 256, 384].  I tried doing this but i get an error
dpv_refined_predicted = F.interpolate(dpv_refined_predicted, size=[256,256,384], mode='linear')

dpv_refined_predicted = F.upsample(dpv_refined_predicted, size=[256,256,384], mode='linear')

NotImplementedError: Got 4D input, but linear mode needs 3D input