No gradients provided for any variable - Custom loss function with random weights depending on the Softmax output

Question

I have difficulties writing a custom loss function that makes use of some random weights generated according to the class/state predicted by the Softmax output. The desired prop