

``mx.nd.SpatialTransformer``
========================================================

Description
----------------------

Applies a spatial transformer to input feature map.


Arguments
------------------

+----------------------------------------+------------------------------------------------------------+
| Argument                               | Description                                                |
+========================================+============================================================+
| ``data``                               | NDArray-or-Symbol.                                         |
|                                        |                                                            |
|                                        | Input data to the SpatialTransformerOp.                    |
+----------------------------------------+------------------------------------------------------------+
| ``loc``                                | NDArray-or-Symbol                                          |
|                                        | localisation net, the output dim should be 6 when          |
|                                        | transform_type is affine. You shold initialize the weight  |
|                                        | and bias with identity                                     |
|                                        | tranform.                                                  |
+----------------------------------------+------------------------------------------------------------+
| ``target.shape``                       | Shape(tuple), optional, default=[0,0].                     |
|                                        |                                                            |
|                                        | output shape(h, w) of spatial transformer: (y, x)          |
+----------------------------------------+------------------------------------------------------------+
| ``transform.type``                     | {'affine'}, required.                                      |
|                                        |                                                            |
|                                        | transformation type                                        |
+----------------------------------------+------------------------------------------------------------+
| ``sampler.type``                       | {'bilinear'}, required.                                    |
|                                        |                                                            |
|                                        | sampling type                                              |
+----------------------------------------+------------------------------------------------------------+
| ``cudnn.off``                          | boolean or None, optional, default=None                    |
|                                        | whether to turn cudnn off                                  |
+----------------------------------------+------------------------------------------------------------+

Value
----------

``out`` The result mx.ndarray