"""
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)