""" BCELossWithMask继承了BCELoss习惯，有相同的对输入形状的要求（batch_num，class_num，image.shape） 且没有进行softmax """ class BCELossWithMask(nn.Module): def __init__(self): super(BCELossWithMask, self).__init__() def forward(self, output, target, mask): loss=torch.tensor(0)