mx.nd.SpatialTransformer
¶
Description¶
Applies a spatial transformer to input feature map.
Arguments¶
Argument |
Description |
---|---|
|
NDArray-or-Symbol. Input data to the SpatialTransformerOp. |
|
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. |
|
Shape(tuple), optional, default=[0,0]. output shape(h, w) of spatial transformer: (y, x) |
|
{‘affine’}, required. transformation type |
|
{‘bilinear’}, required. sampling type |
|
boolean or None, optional, default=None whether to turn cudnn off |
Value¶
out
The result mx.ndarray