Function vector_norm

pub fn vector_norm<B, const D: usize>(
    x: Tensor<B, D>,
    norm: impl Into<Norm>,
    dim: usize,
) -> Tensor<B, D>
where
    B: Backend,

Computes the vector norm of a tensor along a specified dimension.

Generic dispatch wrapper over specialized / optimized norms.

See:

• torch.linalg.vector_norm
• numpy.linalg.vector_norm

Arguments

• x - The input tensor.
• norm - The selected norm.
• dim - The dimension to compute the norm over.

Returns

The vector norm of the input tensor.