bartorch.normalize

bartorch.normalize(input, axes, *, l1=False)

Divide by the l2 norm over axes, or the l1 norm (-b), per index of the other axes.