Fancy indexing in tensorflow

Question

I have implemented a 3D CNN with a custom loss function (Ax\' - y)^2 where x\' is a flattened and cropped vector of the 3D output from the CNN, y is the ground