Convert tensor of floats to tensor fo integers inside loss function

Question

I\'m building a loss function, and I need to use the y_true and y_pred as indexes for a matrix which I\'m using to calculate the loss. The problem is, both of these come as floa