Function normalize 

pub fn normalize(
    input: &Tensor,
    p: f64,
    dim: i64,
    eps: f64,
    out: Option<&mut Tensor>,
) -> Result<Tensor>

Normalize tensor using Lp norm