Class Swish

modules/se.py:5–7

__all__ = ['SE3d']

class Swish(nn.Module):
    def forward(self,x):
        return  x * torch.sigmoid(x)
class SE3d(nn.Module):
    def __init__(self, channel, reduction=8, use_relu=False):
        super().__init__()