magnet.graph_utils.randomrotate

magnet.graph_utils.randomrotate(coords_sample: Tensor) → Tensor

Randomly rotate the mesh.

Parameters:

coords_sample (torch.Tensor) – Coordinates of the mesh to be rotated.

Returns:

The rotated coordinates.

Return type:

torch.Tensor

Notes

Rotation is always computed on GPU if available.