Tensorflow loss function to find difference in index

Question

So I have been working on a convolution neural network that is trying to pick a row out of time series data. I ended my network with a dense layer with the same number of output