When I calculate the default 2-norm of matrix on 2 dimensions, it works.

However, when I calculate the 1-norm on these 2 dimensions, it fails.(Just add ‘p=1’)

image.png859×339 9.79 KB

The error is so evident, the dim here must be a int.
The 1-norm of matrix is the maximum-column sum of the matrix, see this[1].

My input “a” and “b” are both 10x32x32 tensor, calculating the 1-norm of last two dimension, what I expect to get is a 10x1x1 tensor.
reference:
[1] Matlab Matrix 1-norm 3