pub fn l2_normalize<B: Backend>(x: Tensor<B, 2>) -> Tensor<B, 2>
L2-normalise each row of (B, D) to unit norm.
(B, D)