improve linalg.norm

[mtar]

rewrite linalg.norm such that it supports different norms.

Reference:
https://numpy.org/doc/stable/reference/generated/numpy.linalg.norm.html
https://pytorch.org/docs/stable/generated/torch.linalg.norm.html#torch.linalg.norm
https://data-apis.org/array-api/latest/extensions/linear_algebra_functions.html#linalg-norm-x-axis-none-keepdims-false-ord-none

Additional functions especially for matrix norm and vector norm possible:
https://pytorch.org/docs/stable/generated/torch.linalg.vector_norm.html#torch.linalg.vector_norm
https://pytorch.org/docs/stable/generated/torch.linalg.matrix_norm.html#torch.linalg.matrix_norm