def forward(self, feature):
    epsilon = 1e-06
    norm = torch.pow((torch.sum(torch.pow(feature, 2), 1) + epsilon), 0.5).unsqueeze(1).expand_as(feature)
    return torch.div(feature, norm)
def forward(self, feature):
    epsilon = 1e-06
    norm = torch.pow((torch.sum(torch.pow(feature, 2), 1) + epsilon), 0.5).unsqueeze(1).expand_as(feature)
    return torch.div(feature, norm)