Ok, I skimmed the paper, but I don’t really understand all aspects.
You feed the output of the model into the “sensing matrix” and somehow get the cell positions?
This part is working already, right?
Does the output of the model have any boundaries, e.g. [-1, 1], so that we could apply a tanh on the last layer?
Since it’s a Linear layer, the values might grow quite big.