如何将CE损失函数变成focal loss 损失函数？

源代码是这样的，CE的损失函数，如何将它变成focal loss的呢？能变吗？求解啊