nvidia.dali.fn.transforms.rotation

nvidia.dali.fn.transforms.rotation(__input=None, /, *, angle, axis=None, bytes_per_sample_hint=[0], center=None, preserve=False, reverse_order=False, seed=-1, device=None, name=None)

Produces a rotation affine transform matrix.

If another transform matrix is passed as an input, the operator applies rotation to the matrix provided.

The number of dimensions is assumed to be 3 if a rotation axis is provided or 2 otherwise.

Note

The output of this operator can be fed directly to CoordTransform and WarpAffine operators.

This operator allows sequence inputs.

Supported backends
  • ‘cpu’

Parameters:

__input (TensorList, optional) – Input to the operator.

Keyword Arguments:
  • angle (float or TensorList of float) –

    Angle, in degrees.

    Supports per-frame inputs.

  • axis (float or list of float or TensorList of float, optional) –

    Axis of rotation (applies only to 3D transforms).

    The vector does not need to be normalized, but it must have a non-zero length.

    Reversing the vector is equivalent to changing the sign of angle.

    Supports per-frame inputs.

  • bytes_per_sample_hint (int or list of int, optional, default = [0]) –

    Output size hint, in bytes per sample.

    If specified, the operator’s outputs residing in GPU or page-locked host memory will be preallocated to accommodate a batch of samples of this size.

  • center (float or list of float or TensorList of float, optional) –

    The center of the rotation.

    If provided, the number of elements should match the dimensionality of the transform.

    Supports per-frame inputs.

  • preserve (bool, optional, default = False) – Prevents the operator from being removed from the graph even if its outputs are not used.

  • reverse_order (bool, optional, default = False) –

    Determines the order when combining affine transforms.

    If set to False (default), the operator’s affine transform will be applied to the input transform. If set to True, the input transform will be applied to the operator’s transform.

    If there’s no input, this argument is ignored.

  • seed (int, optional, default = -1) –

    Random seed.

    If not provided, it will be populated based on the global seed of the pipeline.