Soft dice score

[Ycblue]

Hey :D
I stumbled upon your code while looking up rising transforms!
I think the soft dice score in losses.soft_dice.py should return
1 - torch.mean(frac, dims=1)
instead of just taking the mean so the optimizer can minimize it to zero.
Or is there another reason that it is implemented that way?

Cheers!