利用BCELoss解決Multi-label問題

class torch.nn.BCELoss(weight=None, size_average=True)[source] 計算 target 與 output 之間的二進制交叉熵。 l o s s ( o , t ) = − 1 n ∑ i ( t [ i ] l o g ( o [ i ] ) + ( 1 − t [ i ] ) l o g ( 1 − o [ i ] ) ) loss(o,

>>阅读原文<<