axisL2norm

Normalize a tensor along a given dimension.

Normalize an N-dimensional tensor along a given axis using the L2 (euclidean) norm.

A simple example where N=2 that illustrates what this means:

Input: [ 3 5 ]
[ 4 12]

Let's say we want to normalize along the 2nd dimension.

Normalization = [ 5 13 ]

Normalized Output = [ 3/5 5/13 ]
[ 4/5 12/13]