burn::tensor::linalg

Function vector_normalize

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

Expand description

Normalize a tensor versus its vector_norm.

Equivalent to x.clone() / vector_norm(x, norm, dim).clamp_min(eps).

§Arguments

x - The input tensor.
norm - The selected norm.
dim - The dimension to compute the norm over.
eps - The epsilon for the norm.

§Returns

The normalized tensor.