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, device=None, name=None)#

生成旋转仿射变换矩阵。

如果将另一个变换矩阵作为输入传递，则运算符会将旋转应用于提供的矩阵。

如果提供了旋转轴，则维度数假定为 3，否则为 2。

注意

此运算符的输出可以直接馈送到 CoordTransform 和 WarpAffine 运算符。

此运算符允许序列输入。

支持的后端

  • ‘cpu’

参数:

__input (TensorList, optional) – 运算符的输入。

关键字参数:

  • angle (float 或 TensorList of float) –

    角度，以度为单位。

    支持 per-frame 输入。

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

    旋转轴（仅适用于 3D 变换）。

    该向量不需要归一化，但必须具有非零长度。

    反转向量等效于更改 angle 的符号。

    支持 per-frame 输入。

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

    输出大小提示，以每个样本的字节数为单位。

    如果指定，则驻留在 GPU 或页锁定主机内存中的运算符输出将被预先分配，以适应此大小的样本批次。

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

    旋转中心。

    如果提供，则元素数量应与变换的维度匹配。

    支持 per-frame 输入。

  • preserve (bool, optional, default = False) – 即使运算符的输出未使用，也阻止从图中删除该运算符。

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

    确定组合仿射变换时的顺序。

    如果设置为 False（默认），则运算符的仿射变换将应用于输入变换。 如果设置为 True，则输入变换将应用于运算符的变换。

    如果没有输入，则忽略此参数。

在本页中