# Copyright (c) OpenMMLab. All rights reserved. import torch def clip_sigmoid(x, eps=1e-4): """Sigmoid function for input feature. Args: x (torch.Tensor): Input feature map with the shape of [B, N, H, W]. eps (float, optional): Lower bound of the range to be clamped to. Defaults to 1e-4. Returns: torch.Tensor: Feature map after sigmoid. """ y = torch.clamp(x.sigmoid_(), min=eps, max=1 - eps) return y