soft_tensor = torch.ones_like(soft_mask)