torch_geometric.transforms.RandomRotate

class RandomRotate(degrees: Union[Tuple[float, float], float], axis: int = 0)[source]

Bases: BaseTransform

Rotates node positions around a specific axis by a randomly sampled factor within a given interval (functional name: random_rotate).

Parameters:
  • degrees (tuple or float) – Rotation interval from which the rotation angle is sampled. If degrees is a number instead of a tuple, the interval is given by \([-\mathrm{degrees}, \mathrm{degrees}]\).

  • axis (int, optional) – The rotation axis. (default: 0)