Focal loss: ValueError: batch_dims = 1 must be less than rank(indices) = 1

Question

I\'m trying to re-implement the focal loss function from https://github.com/artemmavrin/focal-loss/ using Tensorflow1.14, but I\'m running into the following error while tes