def _normalize(tensor, norm_layer):
    '\n    Broadcast layer norm\n    '
    size = tensor.size()
    return norm_layer(tensor.view((- 1), size[(- 1)])).view(size)
Broadcast layer norm
def _normalize(tensor, norm_layer):
    '\n    \n    '
    size = tensor.size()
    return norm_layer(tensor.view((- 1), size[(- 1)])).view(size)