I am trying to implement a CNN that receives as input a 640 by 368 sized tensor where only the center 30 columns (i.e. columns 169 to 199) keep their real values and everything else is set to 0 and outputs a binarized (1s and 0s) tensor of size 338 essentially predicting what column indices in the input that were set to 0 are of importance.