Function tch_utils::ops_2d::normalize_2d

pub fn normalize_2d(a: &Tensor) -> Tensor

Normalize vectors in 2d pictures to unit length.

Arguments

a: Tensor - The tensor to normalize [N, C, H, W]

Returns

Tensor - The normalized tensor [N, C, H, W]