8 lines (8 with data), 749 Bytes

B


`ÉaSã@sdd
lZddl
Tdd„Zd
S)éNé
)
Ú
*cCsÆ|d
k}|
t 
|
¡
k}
t|ƒ
}t|
ƒ
}
||
}tj|dd}tj|dd}||}tj|dd}tj|dd}|
|
}tj|dd}tj|dd}d|||d}ddt |¡
| d¡
}|S)zw
    2d dice loss
    :param input: predict tensor
    :param target: target tensor
    :return: scalar loss value
    gà?é)
Údimgíµ ÷Æ°>rr
)ÚtorchÚmaxZto_float_and_cudaÚsumÚsize)ÚinputÚtargetÚnumZden1Zden2ÚdiceZ
dice_total©rú2/home/wangqiuli/Code/Newcode1222_monk/loss/dice.pyÚ	dice_losss 








r)rÚutilsrrrrrÚ<module>
s