Error when using dice loss function for multiclass segmentation

Question

I got the following error on the line where we calculate the intersection variable: InvalidArgumentError: cannot compute Mul as input #1(zero-based) was expected to be a int