blob: 209d7ea8bc59a2e447481353fd84987eaadbe88d [file] [log] [blame]
``mx.nd.GridGenerator``
==============================================
Description
----------------------
Generates 2D sampling grid for bilinear sampling.
Arguments
------------------
+----------------------------------------+------------------------------------------------------------+
| Argument | Description |
+========================================+============================================================+
| ``data`` | NDArray-or-Symbol. |
| | |
| | Input data to the function. |
+----------------------------------------+------------------------------------------------------------+
| ``transform.type`` | {'affine', 'warp'}, required. |
| | |
| | The type of transformation. For `affine`, input data |
| | should be an affine matrix of size (batch, 6). For `warp`, |
| | input data should be an optical flow of size (batch, 2, h, |
| | w). |
+----------------------------------------+------------------------------------------------------------+
| ``target.shape`` | Shape(tuple), optional, default=[0,0]. |
| | |
| | Specifies the output shape (H, W). This is required if |
| | transformation type is `affine`. If transformation type is |
| | `warp`, this parameter is |
| | ignored. |
+----------------------------------------+------------------------------------------------------------+
Value
----------
``out`` The result mx.ndarray