Add docstring to spatial transformer Ops

theano/gpuarray/dnn.py

@@ -2740,8 +2740,7 @@ def dnn_batch_normalization_test(inputs, gamma, beta, mean, var,
 
 class GpuDnnTransformerDesc(COp):
     """
-    This Op builds a spatial transformer descriptor for use in spatial transformer network
-    operations.
+    Descriptor Op for cuDNN Spatial Transformer.
     """
     __props__ = ('precision',)
     params_type = ParamsType(precision=cudnn.cudnnDataType_t)
@@ -2794,6 +2793,9 @@ class GpuDnnTransformerDesc(COp):
 
 
 class GpuDnnTransformerGrid(DnnBase):
+    """
+    Grid generator Op for cuDNN Spatial Transformer.
+    """
     __props__ = ()
     _cop_num_inputs = 3
     _cop_num_outputs = 1
@@ -2841,6 +2843,9 @@ class GpuDnnTransformerGrid(DnnBase):
 
 
 class GpuDnnTransformerSampler(DnnBase):
+    """
+    Grid sampler Op for cuDNN Spatial Transformer.
+    """
     __props__ = ()
     _cop_num_inputs = 3
     _cop_num_outputs = 1
@@ -2891,7 +2896,7 @@ class GpuDnnTransformerSampler(DnnBase):
 
 class GpuDnnTransformerGradI(DnnBase):
     """
-    Gradients of inputs of the spatial transformer
+    Gradient of inputs Op for cuDNN Spatial Transformer.
     """
     __props__ = ()
     _cop_num_inputs = 4
@@ -2936,7 +2941,7 @@ class GpuDnnTransformerGradI(DnnBase):
 
 class GpuDnnTransformerGradT(DnnBase):
     """
-    Gradients of the affine transformation generated by the localisation network
+    Gradient of affine transformations Op for cuDNN Spatial Transformer.
     """
     __props__ = ()
     _cop_num_inputs = 2