[cf6a9e]: / loss / __pycache__ / dice.cpython-35.pyc

Download this file

12 lines (12 with data), 851 Bytes



”¡£_Sã@s&ddlZddlTdd„ZdS)éNé)Ú*cCs|dk}|tj|ƒk}t|ƒ}t|ƒ}||}tj|ddƒ}tj|ddƒ}||}tj|ddƒ}tj|ddƒ}||}tj|ddƒ}tj|ddƒ}d|||d}ddtj|ƒ|jdƒ}|S)zw
    2d dice loss
    :param input: predict tensor
    :param target: target tensor
    :return: scalar loss value
    gà?Údimégíµ ÷ư>rr)ÚtorchÚmaxZto_float_and_cudaÚsumÚsize)ÚinputÚtargetÚnumÚden1Zden2ÚdiceZ
dice_total©rúG/raid/data/wangqiuli/Documents/YH_Interpretable/Version1/losses/dice.pyÚ	dice_losss 


$r)rÚutilsrrrrrÚ<module>s