src/transformers/models/yoso/modeling_yoso.py/normalize
def normalize(input_tensors):
    if type(input_tensors) is list:
        out = []
        for tensor in input_tensors:
            out.append(nn.functional.normalize(tensor, p=2, dim=-1))
        return out
    else:
        return nn.functional.normalize(input_tensors, p=2, dim=-1)