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
andWarpAffine
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.