def build(self, mask_tensor):
    loss = flow.sum(self.m(mask_tensor))
    loss.backward()
    return loss
def build(self, mask_tensor):
    loss = flow.sum(self.m(mask_tensor))
    loss.backward()
    return loss