create an heatmap image using the coordinate points in tensorflow, while keeping the gradients

Question

I have a tensor with the shape cp = [30*2]. The tensor is created by my CNN. The predicted 60 points are the coordinate points of a heatmap I wanna create. The