Image level Dice

[balajiselvaraj1601]

I would like to capture image level dice score
label and predictions are of [batch_sz, 1, im_sz, im_sz] dimensions.

What's the proper setting to compute dice score for each item in the batch