Function tch_utils::ops_2d::rotate_2d

pub fn rotate_2d(t: &Tensor, angle: f64) -> Tensor

Rotate the of vectors in 2d pictures.

Arguments

t: Tensor - The tensor to rotate [N, 2, H, W] angle: f64 - The angle of rotation in radians

Returns

Tensor - The rotated tensor [N, 2, H, W]