mindspore.ops.AffineGrid

class mindspore.ops.AffineGrid(align_corners=False)

基于一批仿射矩阵 theta 生成一个2D 或 3D 的流场（采样网格）。

警告

这是一个实验性API，后续可能修改或删除。

更多参考详见 mindspore.ops.affine_grid() 。

参数：

• align_corners (bool，可选) - 在几何上，我们将输入的像素视为正方形而不是点。如果设置为 True ，则极值 -1 和 1 指输入像素的中心。如果设置为 False ，则极值 -1 和 1 指输入像素的边角，从而使采样与分辨率无关。默认值： False 。

输入：

• theta (Tensor) - 仿射矩阵输入，其shape为 \((N, 2, 3)\) 用于 2D grid 或 \((N, 3, 4)\) 用于 3D grid。

• output_size (tuple[int]) - 目标输出图像大小。指格式为 \((N, C, H, W)\) 的2D grid或 \((N, C, D, H, W)\) 的3D grid的大小。

输出：

Tensor，其数据类型与 theta 相同，其shape为 \((N, H, W, 2)\) 用于 2D grid或 \((N, D, H, W, 3)\) 用于 3D grid。

支持平台：

Ascend GPU CPU

样例：

>>> affinegrid = ops.AffineGrid(align_corners=False)
>>> theta = Tensor([[[0.8, 0.5, 0],[-0.5, 0.8, 0]]], mindspore.float32)
>>> out_size = (1, 3, 2, 3)
>>> output = affinegrid(theta, out_size)
>>> print(output)
[[[[-0.78333336 -0.06666666]
[-0.25       -0.4       ]
[ 0.28333336 -0.73333335]]
[[-0.28333336  0.73333335]
[ 0.25        0.4       ]
[ 0.78333336  0.06666666]]]]