def forward(self, x):
    value = self.smooth_l1_loss(x)
    if (self.reduction == 'sum'):
        loss = value.mean(dim=1)
        loss = loss.sum()
    else:
        loss = value.sum(dim=0)
    return loss
def forward(self, x):
    value = self.smooth_l1_loss(x)
    if (self.reduction == 'sum'):
        loss = value.mean(dim=1)
        loss = loss.sum()
    else:
        loss = value.sum(dim=0)
    return loss